AI-based diagnostic tools