AI-driven diagnostics