https://github.com/Hansong-Zhang/CCC

AAAI 2024, Coupled Confusion Correction: Learning from Crowds with Sparse Annotations
https://github.com/Hansong-Zhang/CCC

Last synced: 9 days ago
JSON representation

AAAI 2024, Coupled Confusion Correction: Learning from Crowds with Sparse Annotations

• Host: GitHub
• URL: https://github.com/Hansong-Zhang/CCC
• Owner: Hansong-Zhang
• Created: 2023-12-12T09:27:45.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2024-04-05T03:30:22.000Z (7 months ago)
• Last Synced: 2024-08-02T15:29:59.794Z (3 months ago)
• Language: Python
• Homepage: https://hansong-zhang.github.io/CCC/
• Size: 2.06 MB
• Stars: 2
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# [AAAI 2024] Coupled Confusion Correction: Learning from Crowds with Sparse Annotations

## [Paper](https://arxiv.org/abs/2312.07331)

## Abstract

As the size of the datasets getting larger, accurately annotating such datasets is becoming more impractical due to the expensiveness on both time and economy. Therefore, crowd-sourcing has been widely adopted to alleviate the cost of collecting labels, which also inevitably introduces label noise and eventually degrades the performance of the model. To learn from crowd-sourcing annotations, modeling the expertise of each annotator is a common but challenging paradigm, because the annotations collected by crowd-sourcing are usually highly-sparse. To alleviate this problem, we propose Coupled Confusion Correction (CCC), where two models are simultaneously trained to correct the confusion matrices learned by each other. Via bi-level optimization, the confusion matrices learned by one model can be corrected by the distilled data from the other. Moreover, we cluster the ``annotator groups'' who share similar expertise so that their confusion matrices could be corrected together. In this way, the expertise of the annotators, especially of those who provide seldom labels, could be better captured. Remarkably, we point out that the annotation sparsity not only means the average number of labels is low, but also there are always some annotators who provide very few labels, which is neglected by previous works when constructing synthetic crowd-sourcing annotations. Based on that, we propose to use Beta distribution to control the generation of the crowd-sourcing labels so that the synthetic annotations could be more consistent with the real-world ones. Extensive experiments are conducted on two types of synthetic datasets and three real-world datasets, the results of which demonstrate that CCC significantly outperforms state-of-the-art approaches.

## Getting Started

We provide the training codes of [CIFAR-10N](http://www.noisylabels.com/) dataset.

1. Create the environment using requirements.yaml

```

conda create --name env_ccc --file requirements.yaml

```

2. Activate the environment

```

conda activate env_ccc

```

3. Training CIFAR-10N

```

python train_ccc.py --dataset=cifar10n --data_path=

```

## Citation

If you find our code useful for your research, please cite our paper.

```

@inproceedings{zhang2024ccc,

      title={Coupled Confusion Correction: Learning from Crowds with Sparse Annotations}, 

      author={Hansong Zhang and Shikun Li and Dan Zeng and Chenggang Yan and Shiming Ge},

      year={2024},

      booktitle={The 38th Annual AAAI Conference on Artificial Intelligence (AAAI)}

}

```