Multi-annotator Deep Learning: A Probabilistic Framework for Classification
About
Solving complex classification tasks using deep neural networks typically requires large amounts of annotated data. However, corresponding class labels are noisy when provided by error-prone annotators, e.g., crowdworkers. Training standard deep neural networks leads to subpar performances in such multi-annotator supervised learning settings. We address this issue by presenting a probabilistic training framework named multi-annotator deep learning (MaDL). A downstream ground truth and an annotator performance model are jointly trained in an end-to-end learning approach. The ground truth model learns to predict instances' true class labels, while the annotator performance model infers probabilistic estimates of annotators' performances. A modular network architecture enables us to make varying assumptions regarding annotators' performances, e.g., an optional class or instance dependency. Further, we learn annotator embeddings to estimate annotators' densities within a latent space as proxies of their potentially correlated annotations. Together with a weighted loss function, we improve the learning from correlated annotation patterns. In a comprehensive evaluation, we examine three research questions about multi-annotator supervised learning. Our findings show MaDL's state-of-the-art performance and robustness against many correlated, spamming annotators.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Classification | DTD | Accuracy47.7 | 87 | |
| Classification | Flowers102 | Top-1 Accuracy85.1 | 38 | |
| Text Classification | AGNews | Accuracy78 | 34 | |
| Classification | labelme | Accuracy86.5 | 30 | |
| Classification | Letter | Accuracy71.2 | 29 | |
| Classification | mgc | Accuracy72.4 | 26 | |
| Classification | aloi | Classification Accuracy79.2 | 26 | |
| Classification | TREC6 | Accuracy91.1 | 26 | |
| Classification | CIFAR-100N | Accuracy42.8 | 26 | |
| Classification | CIFAR-10N | Accuracy80.5 | 26 |