AI-driven quality assurance