DUEL: Duplicate Elimination on Active Memory for Self-Supervised Class-Imbalanced Learning
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2024
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866929243141701632 |
|---|---|
| author | Choi, Won-Seok Lee, Hyundo Han, Dong-Sig Park, Junseok Koo, Heeyeon Zhang, Byoung-Tak |
| author_facet | Choi, Won-Seok Lee, Hyundo Han, Dong-Sig Park, Junseok Koo, Heeyeon Zhang, Byoung-Tak |
| contents | Recent machine learning algorithms have been developed using well-curated datasets, which often require substantial cost and resources. On the other hand, the direct use of raw data often leads to overfitting towards frequently occurring class information. To address class imbalances cost-efficiently, we propose an active data filtering process during self-supervised pre-training in our novel framework, Duplicate Elimination (DUEL). This framework integrates an active memory inspired by human working memory and introduces distinctiveness information, which measures the diversity of the data in the memory, to optimize both the feature extractor and the memory. The DUEL policy, which replaces the most duplicated data with new samples, aims to enhance the distinctiveness information in the memory and thereby mitigate class imbalances. We validate the effectiveness of the DUEL framework in class-imbalanced environments, demonstrating its robustness and providing reliable results in downstream tasks. We also analyze the role of the DUEL policy in the training process through various metrics and visualizations. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_08963 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | DUEL: Duplicate Elimination on Active Memory for Self-Supervised Class-Imbalanced Learning Choi, Won-Seok Lee, Hyundo Han, Dong-Sig Park, Junseok Koo, Heeyeon Zhang, Byoung-Tak Machine Learning Artificial Intelligence Recent machine learning algorithms have been developed using well-curated datasets, which often require substantial cost and resources. On the other hand, the direct use of raw data often leads to overfitting towards frequently occurring class information. To address class imbalances cost-efficiently, we propose an active data filtering process during self-supervised pre-training in our novel framework, Duplicate Elimination (DUEL). This framework integrates an active memory inspired by human working memory and introduces distinctiveness information, which measures the diversity of the data in the memory, to optimize both the feature extractor and the memory. The DUEL policy, which replaces the most duplicated data with new samples, aims to enhance the distinctiveness information in the memory and thereby mitigate class imbalances. We validate the effectiveness of the DUEL framework in class-imbalanced environments, demonstrating its robustness and providing reliable results in downstream tasks. We also analyze the role of the DUEL policy in the training process through various metrics and visualizations. |
| title | DUEL: Duplicate Elimination on Active Memory for Self-Supervised Class-Imbalanced Learning |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2402.08963 |