AI-assisted diagnosis