AI powered defect classification