Learning underrepresented classes from decentralized partially labeled medical images
Using decentralized data for federated training is one promising emerging research direction for alleviating data scarcity in the medical domain. However, in contrast to large-scale fully labeled data commonly seen in general object recognition tasks, the local medical datasets are more likely to on...
| Main Authors: | , , |
|---|---|
| Format: | Conference item |
| Language: | English |
| Published: |
Springer
2022
|
| _version_ | 1797110696497905664 |
|---|---|
| author | Dong, N Kampffmeyer, M Voiculescu, I |
| author_facet | Dong, N Kampffmeyer, M Voiculescu, I |
| author_sort | Dong, N |
| collection | OXFORD |
| description | Using decentralized data for federated training is one promising emerging research direction for alleviating data scarcity in the medical domain. However, in contrast to large-scale fully labeled data commonly seen in general object recognition tasks, the local medical datasets are more likely to only have images annotated for a subset of classes of interest due to high annotation costs. In this paper, we consider a practical yet under-explored problem, where underrepresented classes only have few labeled instances available and only exist in a few clients of the federated system. We show that standard federated learning approaches fail to learn robust multi-label classifiers with extreme class imbalance and address it by proposing a novel federated learning framework, FedFew. FedFew consists of three stages, where the first stage leverages federated self-supervised learning to learn class-agnostic representations. In the second stage, the decentralized partially labeled data are exploited to learn an energy-based multi-label classifier for the common classes. Finally, the underrepresented classes are detected based on the energy and a prototype-based nearest-neighbor model is proposed for few-shot matching. We evaluate FedFew on multi-label thoracic disease classification tasks and demonstrate that it outperforms the federated baselines by a large margin. |
| first_indexed | 2024-03-07T07:58:27Z |
| format | Conference item |
| id | oxford-uuid:0b716548-8ede-4161-856b-d7a83d4b0438 |
| institution | University of Oxford |
| language | English |
| last_indexed | 2024-03-07T07:58:27Z |
| publishDate | 2022 |
| publisher | Springer |
| record_format | dspace |
| spelling | oxford-uuid:0b716548-8ede-4161-856b-d7a83d4b04382023-09-18T07:36:22ZLearning underrepresented classes from decentralized partially labeled medical imagesConference itemhttp://purl.org/coar/resource_type/c_5794uuid:0b716548-8ede-4161-856b-d7a83d4b0438EnglishSymplectic ElementsSpringer2022Dong, NKampffmeyer, MVoiculescu, IUsing decentralized data for federated training is one promising emerging research direction for alleviating data scarcity in the medical domain. However, in contrast to large-scale fully labeled data commonly seen in general object recognition tasks, the local medical datasets are more likely to only have images annotated for a subset of classes of interest due to high annotation costs. In this paper, we consider a practical yet under-explored problem, where underrepresented classes only have few labeled instances available and only exist in a few clients of the federated system. We show that standard federated learning approaches fail to learn robust multi-label classifiers with extreme class imbalance and address it by proposing a novel federated learning framework, FedFew. FedFew consists of three stages, where the first stage leverages federated self-supervised learning to learn class-agnostic representations. In the second stage, the decentralized partially labeled data are exploited to learn an energy-based multi-label classifier for the common classes. Finally, the underrepresented classes are detected based on the energy and a prototype-based nearest-neighbor model is proposed for few-shot matching. We evaluate FedFew on multi-label thoracic disease classification tasks and demonstrate that it outperforms the federated baselines by a large margin. |
| spellingShingle | Dong, N Kampffmeyer, M Voiculescu, I Learning underrepresented classes from decentralized partially labeled medical images |
| title | Learning underrepresented classes from decentralized partially labeled medical images |
| title_full | Learning underrepresented classes from decentralized partially labeled medical images |
| title_fullStr | Learning underrepresented classes from decentralized partially labeled medical images |
| title_full_unstemmed | Learning underrepresented classes from decentralized partially labeled medical images |
| title_short | Learning underrepresented classes from decentralized partially labeled medical images |
| title_sort | learning underrepresented classes from decentralized partially labeled medical images |
| work_keys_str_mv | AT dongn learningunderrepresentedclassesfromdecentralizedpartiallylabeledmedicalimages AT kampffmeyerm learningunderrepresentedclassesfromdecentralizedpartiallylabeledmedicalimages AT voiculescui learningunderrepresentedclassesfromdecentralizedpartiallylabeledmedicalimages |