Command Palette
Search for a command to run...
SubRegWeigh: Effective and Efficient Annotation Weighing with Subword Regularization
Kohei Tsuji; Tatsuya Hiraoka; Yuchang Cheng; Tomoya Iwakura

Abstract
NLP datasets may still contain annotation errors, even when they are manually annotated. Researchers have attempted to develop methods to automatically reduce the adverse effect of errors in datasets. However, existing methods are time-consuming because they require many trained models to detect errors. This paper proposes a time-saving method that utilizes a tokenization technique called subword regularization to simulate multiple error detection models for detecting errors. Our proposed method, SubRegWeigh, can perform annotation weighting four to five times faster than the existing method. Additionally, SubRegWeigh improved performance in document classification and named entity recognition tasks. In experiments with pseudo-incorrect labels, SubRegWeigh clearly identifies pseudo-incorrect labels as annotation errors. Our code is available at https://github.com/4ldk/SubRegWeigh .
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| named-entity-recognition-ner-on-conll-2003 | RoBERTa + SubRegWeigh (K-means) | F1: 93.81 |
| named-entity-recognition-ner-on-conll-2003 | LUKE + SubRegWeigh (K-means) | F1: 94.2 |
| named-entity-recognition-ner-on-conll-2020 | RoBERTa + SubRegWeigh (K-means) | F1: 94.96 |
| named-entity-recognition-ner-on-conll-2020 | LUKE + SubRegWeigh (K-means) | F1: 95.31 |
| named-entity-recognition-on-conll | LUKE + SubRegWeigh (K-means) | F1: 95.27 |
| named-entity-recognition-on-conll | RoBERTa + SubRegWeigh (K-means) | F1: 95.45 |
| named-entity-recognition-on-wnut-2017 | RoBERTa + SubRegWeigh (K-means) | F1: 60.29 |
| semantic-textual-similarity-on-mrpc | RoBERTa + SubRegWeigh (K-means) | Accuracy: 86.82% |
| sentiment-analysis-on-sst-2-binary | RoBERTa + SubRegWeigh (K-means) | Accuracy: 94.84 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.