MedNLI Is Not Immune: Natural Language Inference Artifacts in the Clinical Domain
Paper · 2021 · Christine Herlihy and Rachel Rudinger
Summary
Large crowd-built inference benchmarks such as SNLI and MultiNLI contain annotation artifacts: label-correlated cues that let a model guess the label from the hypothesis alone. Herlihy and Rudinger test whether MedNLI, written by physicians over premises from MIMIC-III clinical notes, avoids this problem. It does not. A premise-blind fastText classifier scores about 63–65 micro-F1, against 33.3 for chance on this class-balanced set. The authors then examine which words drive each label, using PMI over scispaCy entity-merged tokens. They test three suspected physician writing habits with χ² tests over MeSH-linked entities:
- hypernyms for entailment,
- plausible causes or co-occurring conditions for neutral,
- negation or assertions of good health for contradiction.
Next, they apply AFLite adversarial filtering to split MedNLI into easy and difficult partitions, and they release the partition ids. On the difficult part, both hypothesis-only and full-input models lose substantial accuracy. The takeaway is that expert annotation alone does not remove shortcuts. Knowledge-heavy domains need different dataset designs, and reported scores should be discounted for artifacts.
Key points
- MedNLI has 11,232 train, 1395 dev and 1422 test pairs. Each premise received one hypothesis per class (entailment, neutral, contradiction), so the label distribution is balanced.
- The hypothesis-only fastText model reaches 64.8 (dev) and 62.6 (test) micro-F1, against a 33.3 baseline. It does best on contradiction (P 73.2 / R 72.8) and worst on entailment (P 56.7 / R 53.8), and it often confuses entailment with neutral.
- The top PMI tokens (add-50 smoothing, minimum count 5) differ by class:
- Entailment: generic terms and modifiers such as comorbidities, pressors, high_risk.
- Neutral: common conditions and behaviors such as smoking, obesity.
- Contradiction: tokens like no_history_of_cancer and normal_heart_rhythm.
- Unlike in SNLI, hypothesis length does not separate the classes (mean about 5 tokens for every label).
- AFLite settings were 64 logistic regression models, m = 5620, k = 500, over averaged MIMIC-III fastText embeddings. On the difficult partition:
- The hypothesis-only model falls to 0.46 dev / 0.40 test.
- The premise-plus-hypothesis model falls from 0.53/0.51 to 0.43/0.40.
Interesting ideas
- I1. A premise-blind classifier works as a cheap probe for dataset shortcuts, because any score above chance shows that labels leak through surface wording.
- I2. Collapsing multi-token entities into single underscore-joined units before counting statistics keeps meaningful spans like normal_heart_rhythm intact.
- I3. Class-specific PMI with heavy smoothing and a frequency floor turns a vague suspicion of bias into a ranked, inspectable list of cue words per label.
- I4. The authors write explicit annotator heuristics (hypernym, probable cause, assertion of good health) as premise-hypothesis patterns and test them with χ² over linked ontology entities, which makes artifacts explainable instead of merely detected.
- I5. Adversarial filtering into easy and difficult partitions, with partition ids published, gives a harder evaluation slice without new annotation.
For Mova
How these ideas could serve Mova, our small open language model — its files, stages and tests:
- I1 (learn): Train a deliberately crippled model on every benchmark Mova uses (e.g. answer-only or question-only) as a negative control, and report Mova's margin over it rather than raw accuracy.
- I2 (tokens): Add an optional entity-merge view of tokens for corpus statistics, so PMI or frequency analyses over Mova's data treat named spans as one unit.
- I3 (learn): Run the same PMI audit on silver data from the witness parsers or LLM teachers, to spot words that a teacher tier systematically ties to one relation or label.
- I4 (world): Using WordNet hypernymy in the knowledge layer, flag entailment items where the "inference" is just a super-type substitution, so the answer readers can be scored separately on trivial versus genuine inferences.
- I5 (answer): Split school-test and fable-QA sets with a simple-model ensemble filter, and track Mova's accuracy on the difficult slice as the headline number.
Institutions
📚 Shelves 6The field of science this work belongs to (OpenAlex fields).Computer Science · 9,086 worksArtificial Intelligence · 8,889 worksArts and Humanities · 9,296 worksLanguage and Linguistics · 9,206 worksBiochemistry, Genetics and Molecular Biology · 116 worksMolecular Biology · 114 works
🧭 Topics 1Research subjects and languages the work is about.Reasoning · 510 works
Cite
Cite as Christine Herlihy and Rachel Rudinger. 2021. MedNLI Is Not Immune: Natural Language Inference Artifacts in the Clinical Domain. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 2: Short Papers).
@inproceedings{herlihy2021mednli,
title = {{MedNLI Is Not Immune: Natural Language Inference Artifacts in the Clinical Domain}},
author = {Christine Herlihy and Rachel Rudinger},
booktitle = {Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 2: Short Papers)},
year = {2021},
doi = {10.18653/v1/2021.acl-short.129},
url = {https://doi.org/10.18653/v1/2021.acl-short.129}
}References
- Samuel R. Bowman, Gabor Angeli, Christopher Potts, and Christopher D. Manning. 2015. A large annotated corpus for learning natural language inference. In Proceedings of the 2015 Conference on Empirical Methods in Natural Language Processing, pages 632–642, Lisbon, Portugal. Association for Computational Linguistics. ↗
- Ronan Le Bras, Swabha Swayamdipta, Chandra Bhagavatula, Rowan Zellers, Matthew E. Peters, Ashish Sabharwal, and Yejin Choi. 2020. Adversarial filters of dataset biases. ↗
- A. L. Goldberger, L. A. N. Amaral, L. Glass, J. M. Hausdorff, P. Ch. Ivanov, R. G. Mark, J. E. Mietus, G. B. Moody, C.-K. Peng, and H. E. Stanley. 2000 (June 13). PhysioBank, PhysioToolkit, and PhysioNet: Components of a new research resource for complex physiologic signals. Circulation, 101(23):e215–e220. Circulation Electronic Pages: http://circ.ahajournals.org/content/101/23/e215.full PMID:1085218; doi: 10.1161/01.CIR.101.23.e215. ↗
- Suchin Gururangan, Swabha Swayamdipta, Omer Levy, Roy Schwartz, Samuel Bowman, and Noah A. Smith. 2018. Annotation artifacts in natural language inference data. In Proceedings of the 2018 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 2 (Short Papers), pages 107–112, New Orleans, Louisiana. Association for Computational Linguistics. ↗
- Chien-Ju Ho, Aleksandrs Slivkins, Siddharth Suri, and Jennifer Wortman Vaughan. 2015. Incentivizing high quality crowdwork. In Proceedings of the 24th International Conference on World Wide Web, WWW ’15, page 419–429, Republic and Canton of Geneva, CHE. International World Wide Web Conferences Steering Committee. ↗
- Alistair EW Johnson, Tom J Pollard, Lu Shen, Liwei H Lehman, Mengling Feng, Mohammad Ghassemi, Benjamin Moody, Peter Szolovits, Leo Anthony Celi, and Roger G Mark. 2016. MIMIC-III, a freely accessible critical care database. Scientific data, 3:160035.
- Armand Joulin, Edouard Grave, Piotr Bojanowski, and Tomas Mikolov. 2016. Bag of tricks for efficient text classification. arXiv preprint arXiv:1607.01759. ↗
- Yang Liu and Yiling Chen. 2017. Machine-learning aided peer prediction. In Proceedings of the 2017 ACM Conference on Economics and Computation, EC ’17, page 63–80, New York, NY, USA. Association for Computing Machinery. find
- Tom McCoy, Ellie Pavlick, and Tal Linzen. 2019. Right for the wrong reasons: Diagnosing syntactic heuristics in natural language inference. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pages 3428–3448, Florence, Italy. Association for Computational Linguistics. ↗
- Mark Neumann, Daniel King, Iz Beltagy, and Waleed Ammar. 2019a. ScispaCy: Fast and Robust Models for Biomedical Natural Language Processing. In Proceedings of the 18th BioNLP Workshop and Shared Task, pages 319–327, Florence, Italy. Association for Computational Linguistics.
- Mark Neumann, Daniel King, Iz Beltagy, and Waleed Ammar. 2019b. ScispaCy: Fast and robust models for biomedical natural language processing. In Proceedings of the 18th BioNLP Workshop and Shared Task, pages 319–327, Florence, Italy. Association for Computational Linguistics.
- Adam Poliak, Jason Naradowsky, Aparajita Haldar, Rachel Rudinger, and Benjamin Van Durme. 2018. Hypothesis only baselines in natural language inference. In Proceedings of the Seventh Joint Conference on Lexical and Computational Semantics, pages 180–191, New Orleans, Louisiana. Association for Computational Linguistics. ↗
- Alexey Romanov and Chaitanya Shivade. 2018. Lessons from Natural Language Inference in the Clinical Domain. CoRR, abs/1808.06752. ↗
- Keisuke Sakaguchi, Ronan Le Bras, Chandra Bhagavatula, and Yejin Choi. 2020. Winogrande: An adversarial winograd schema challenge at scale. Proceedings of the AAAI Conference on Artificial Intelligence, 34(05):8732–8740. find
- Masatoshi Tsuchiya. 2018. Performance impact caused by hidden bias of training data for recognizing textual entailment. In Proceedings of the Eleventh International Conference on Language Resources and Evaluation (LREC 2018), Miyazaki, Japan. European Language Resources Association (ELRA).
- Adina Williams, Nikita Nangia, and Samuel R. Bowman. 2017. A Broad-Coverage Challenge Corpus for Sentence Understanding through Inference. CoRR, abs/1704.05426. ↗
- Rowan Zellers, Ari Holtzman, Yonatan Bisk, Ali Farhadi, and Yejin Choi. 2019. HellaSwag: Can a machine really finish your sentence? In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, pages 4791– 4800, Florence, Italy. Association for Computational Linguistics. ↗
Updated
This page in Markdown → https://md.research.elamur.ai/free/en/papers/herlihy-2021-mednli-is-not-immune-natural/