Summary
The author proposes a novel method to calibrate models for subjective NLP classification tasks when crowdsourced annotations are available (rather than each instance only having a single ground truth). Here, "calibrate" refers to abstaining from making a prediction when the model is uncertain.
Specifically, a base model is trained on aggregated labels using the usual cross entropy loss (RoBERTa is used in this paper), then a classifier is trained for each worker. The worker classifier is formed by passing the embedding of the CLS token from the base model through a single hidden layer MLP. Subsequently, the output distributions from all worker models are aggregated into a crowd distribution. The distance between this crowd distribution and the base model's output distribution is calculated, and if the distance exceeds a threshold, the model chooses not to output a prediction. The author proposes several methods for how to aggregate worker distributions into a crowd distribution and how to calculate the distance. If the dataset only contains crowdsourced annotations but without worker information, then a crowd model is trained directly to fit each instance's label distribution, and the same method is used to calculate the distance between the crowd distribution and the base model's output distribution.
The author tested the proposed method and other baselines on the tasks of hate speech detection and natural language inference (NLI). The results showed competitive or superior performance.
Reasons to accept
Annotator disagreement in subjective NLP tasks is a prevalent issue that affects many applications.
The paper introduces a novel approach to calibrate models for subjective tasks, emphasizing the value of integrating human annotator insights into the model's decision-making process.
Reasons to reject
The author only conducted experiments on two tasks, where the NLI dataset has far more annotations per instance than typical ones, and the proposed model did not show superior performance to the baseline on the hate speech datasets. Additionally, in Table 2, the author chose different groups of workers to train worker models for each dataset, which also shows that the proposed model is quite sensitive to the choice of workers. In this case, the author should conduct more comprehensive testing of the proposed model on more datasets. For instance, a dataset for detecting online misogyny [1], and a dataset for multilabel emotion classification [2]. Moreover, comparisons should be made with some highly relevant works, such as [3], which also proposed training a model for each worker and discussed the calculation of uncertainty.
* [1] An Expert Annotated Dataset for the Detection of Online Misogyny (EACL 2021)
* [2] GoEmotions: A Dataset of Fine-Grained Emotions (ACL 2020)
* [3] Dealing with Disagreements: Looking Beyond the Majority Vote in Subjective Annotations (ACL 2022)
Although the paper primarily focuses on subjective NLP tasks, the chosen evaluation metrics are still based on the assumption that each instance has only one ground truth, such as accuracy in Cov@Acc=0.9. In fact, while reading the paper, I initially thought the evaluation would be about the ground truth being a distribution, the model also outputs a distribution, and then evaluating the performance by calculating the distance between the two distributions.
Another confusing point is "confidence." For a typical softmax output, like [0.4, 0.6], we would consider it represents p(y=0)=0.4, p(y=1)=0.6, or say the prediction is y=1 with confidence=0.6. However, in subjective tasks, if the ground truth is considered a distribution, then [0.4,0.6] just represents the predicted distribution by the model, which does not contain any information about confidence. It means the model might be very certain that the distribution is [0.4,0.6] or very uncertain. The unknown confidence can be calculated through other means, such as in the proposed method, if the base model's output distribution is very close to the crowd model's output, it is considered high confidence. However, in the paper, the term confidence seems to be interchangeably used with the model's output, as mentioned in the paragraph above Figure 4:
> "If the model confidence is far off from the crowd distribution, we want the model to abstain, i.e. not make a prediction."
But in Sec 4.1.2 it is mentioned:
> "If the model’s output distribution is far off from the human probability distribution, the model will abstain."
Clarifying these concepts more clearly could potentially reduce reader confusion.