Busca avançada
Ano de início
Entree


BiasPruner: Debiased Continual Learning for Medical Image Classification

Texto completo
Autor(es):
Bayasi, Nourhan ; Fayyad, Jamil ; Bissoto, Alceu ; Hamarneh, Ghassan ; Garbi, Rafeef
Número total de Autores: 5
Tipo de documento: Artigo Científico
Fonte: MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2024, PT X; v. 15010, p. 12-pg., 2024-01-01.
Resumo

Continual Learning (CL) is crucial for enabling networks to dynamically adapt as they learn new tasks sequentially, accommodating new data and classes without catastrophic forgetting. Diverging from conventional perspectives on CL, our paper introduces a new perspective wherein forgetting could actually benefit the sequential learning paradigm. Specifically, we present BiasPruner, a CL framework that intentionally forgets spurious correlations in the training data that could lead to shortcut learning. Utilizing a new bias score that measures the contribution of each unit in the network to learning spurious features, BiasPruner prunes those units with the highest bias scores to form a debiased subnetwork preserved for a given task. As BiasPruner learns a new task, it constructs a new debiased subnetwork, potentially incorporating units from previous subnetworks, which improves adaptation and performance on the new task. During inference, BiasPruner employs a simple task-agnostic approach to select the best debiased subnetwork for predictions. We conduct experiments on three medical datasets for skin lesion classification and chest X-Ray classification and demonstrate that BiasPruner consistently outperforms SOTA CL methods in terms of classification performance and fairness. Our code is available here. (AU)

Processo FAPESP: 19/19619-7 - Geração ilimitada de imagens de lesões de pele usando redes generativas adversariais
Beneficiário:Alceu Emanuel Bissoto
Modalidade de apoio: Bolsas no Brasil - Doutorado
Processo FAPESP: 22/09606-8 - Compreensão do papel de atalhos e mudanças de distribuição para generalização de redes neurais profundas
Beneficiário:Alceu Emanuel Bissoto
Modalidade de apoio: Bolsas no Exterior - Estágio de Pesquisa - Doutorado