SafetySecurityAccountabilitycs.LG
Exact Certification of Neural Networks and Partition Aggregation Ensembles against Label Poisoning
Ajinkya Mohgaonkar, Lukas Gosch, Mahalakshmi Sabanayagam, Debarghya Ghoshdastidar, Stephan Günnemann
April 13, 2026
arXiv.org
Key Takeaways
-
Label-flipping attacks pose a serious threat to AI systems: The paper addresses attacks where training data labels are intentionally corrupted to cause models to make wrong predictions, highlighting a real vulnerability in supervised learning that requires formal safeguards.
-
New certification method provides stronger guarantees: EnsembleCert offers tighter robustness guarantees than existing approaches by examining how individual classifiers work (white-box analysis) rather than treating them as black boxes, meaning it can certify that models will resist more label corruption attacks while maintaining accuracy.
-
Significant practical improvements demonstrated: On the CIFAR-10 image dataset, the new method certified protection against 26.5% more label flips compared to previous approaches while using 100 times fewer computational partitions, suggesting the framework is both more effective and more efficient.
-
Makes formal verification of neural networks more practical: By connecting neural networks to kernel methods mathematically, the researchers developed the first exact, polynomial-time calculable certificate for neural networks against label poisoning, making it feasible to formally verify robustness at scale rather than relying on empirical testing alone.
-
Ensemble methods strengthen AI robustness: The work demonstrates that combining multiple classifiers with rigorous mathematical analysis can provide formal guarantees about model behavior under attack, offering a pathway toward more trustworthy AI systems in high-stakes applications.