Class and Region-Adaptive Constraints for Network Calibration

التفاصيل البيبلوغرافية
العنوان: Class and Region-Adaptive Constraints for Network Calibration
المؤلفون: Murugesan, Balamurali, Silva-Rodriguez, Julio, Ayed, Ismail Ben, Dolz, Jose
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computer Vision and Pattern Recognition
الوصف: In this work, we present a novel approach to calibrate segmentation networks that considers the inherent challenges posed by different categories and object regions. In particular, we present a formulation that integrates class and region-wise constraints into the learning objective, with multiple penalty weights to account for class and region differences. Finding the optimal penalty weights manually, however, might be unfeasible, and potentially hinder the optimization process. To overcome this limitation, we propose an approach based on Class and Region-Adaptive constraints (CRaC), which allows to learn the class and region-wise penalty weights during training. CRaC is based on a general Augmented Lagrangian method, a well-established technique in constrained optimization. Experimental results on two popular segmentation benchmarks, and two well-known segmentation networks, demonstrate the superiority of CRaC compared to existing approaches. The code is available at: https://github.com/Bala93/CRac/
Comment: Under review
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2403.12364
رقم الأكسشن: edsarx.2403.12364
قاعدة البيانات: arXiv