AI-assisted diagnostics