Meta-learning Pathologies from Radiology Reports using Variance Aware Prototypical Networks

Arijit Sehanobish, Kawshik Kannan, Nabila Abraham, Anasuya Das, Benjamin Odry


Abstract
Large pretrained Transformer-based language models like BERT and GPT have changed the landscape of Natural Language Processing (NLP). However, fine tuning such models still requires a large number of training examples for each target task, thus annotating multiple datasets and training these models on various downstream tasks becomes time consuming and expensive. In this work, we propose a simple extension of the Prototypical Networks for few-shot text classification. Our main idea is to replace the class prototypes by Gaussians and introduce a regularization term that encourages the examples to be clustered near the appropriate class centroids. Experimental results show that our method outperforms various strong baselines on 13 public and 4 internal datasets. Furthermore, we use the class distributions as a tool for detecting potential out-of-distribution (OOD) data points during deployment.
Anthology ID:
2022.emnlp-industry.34
Volume:
Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing: Industry Track
Month:
December
Year:
2022
Address:
Abu Dhabi, UAE
Editors:
Yunyao Li, Angeliki Lazaridou
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
332–347
Language:
URL:
https://aclanthology.org/2022.emnlp-industry.34
DOI:
10.18653/v1/2022.emnlp-industry.34
Bibkey:
Cite (ACL):
Arijit Sehanobish, Kawshik Kannan, Nabila Abraham, Anasuya Das, and Benjamin Odry. 2022. Meta-learning Pathologies from Radiology Reports using Variance Aware Prototypical Networks. In Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing: Industry Track, pages 332–347, Abu Dhabi, UAE. Association for Computational Linguistics.
Cite (Informal):
Meta-learning Pathologies from Radiology Reports using Variance Aware Prototypical Networks (Sehanobish et al., EMNLP 2022)
Copy Citation:
PDF:
https://aclanthology.org/2022.emnlp-industry.34.pdf