Class and Region-Adaptive Constraints for Network Calibration

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Murugesan, Balamurali, Silva-Rodriguez, Julio, Ayed, Ismail Ben, Dolz, Jose
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911803073626112
author Murugesan, Balamurali
Silva-Rodriguez, Julio
Ayed, Ismail Ben
Dolz, Jose
author_facet Murugesan, Balamurali
Silva-Rodriguez, Julio
Ayed, Ismail Ben
Dolz, Jose
contents 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/
format Preprint
id arxiv_https___arxiv_org_abs_2403_12364
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Class and Region-Adaptive Constraints for Network Calibration
Murugesan, Balamurali
Silva-Rodriguez, Julio
Ayed, Ismail Ben
Dolz, Jose
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/
title Class and Region-Adaptive Constraints for Network Calibration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.12364