AP-OOD: Attention Pooling for Out-of-Distribution Detection
Not on the map. Why?
Abstract (arXiv)
Out-of-distribution (OOD) detection, which maps high-dimensional data into a scalar OOD score, is critical for the reliable deployment of machine learning models. A key challenge in recent research is how to effectively leverage and aggregate token embeddings from language models to obtain the OOD score. In this work, we propose AP-OOD, a novel OOD detection method for natural language that goes beyond simple average-based aggregation by exploiting token-level information. AP-OOD is a semi-supervised approach that flexibly interpolates between unsupervised and supervised settings, enabling the use of limited auxiliary outlier data. Empirically, AP-OOD sets a new state of the art in OOD detection for text: in the unsupervised setting, it reduces the FPR95 (false positive rate at 95% true positives) from 27.84% to 4.67% on XSUM summarization, and from 77.08% to 70.37% on WMT15 En-Fr translation.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 79 of its 111 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
79 of 111 references.
References · 79
- 2017Attention Is All You Needcited by 59,932
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025Qwen3 Technical Reportcited by 8,195
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
38 more
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,308
- 2020Energy-based Out-of-distribution Detectioncited by 1,296
- 2017Get To The Point: Summarization with Pointer-Generator Networkscited by 1,281
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
- 2018Attention-based Deep Multiple Instance Learningcited by 1,077
- 2015Teaching Machines to Read and Comprehendcited by 1,039
- 2018Deep Anomaly Detection with Outlier Exposurecited by 939
- 2021Towards Total Recall in Industrial Anomaly Detectioncited by 772
- 2024Phi-4 Technical Reportcited by 675
- 2022Out-of-Distribution Detection with Deep Nearest Neighborscited by 596
- 2021TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classificationcited by 549
- 2019SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarizationcited by 530
- 2019Using Self-Supervised Learning Can Improve Model Robustness and Uncertaintycited by 511
- 2021ReAct: Out-of-distribution Detection With Rectified Activationscited by 486
- 2019Scaling Out-of-Distribution Detection for Real-World Settingscited by 465
- 2020Hopfield Networks is All You Needcited by 461
- 2020CSI: Novelty Detection via Contrastive Learning on Distributionally Shifted Instancescited by 441
- 2020A Unifying Review of Deep and Shallow Anomaly Detectioncited by 423
- 2022ViM: Out-Of-Distribution with Virtual-logit Matchingcited by 404
- 2021On the Importance of Gradients for Detecting Distributional Shifts in the Wildcited by 338
- 2022VOS: Learning What You Don't Know by Virtual Outlier Synthesiscited by 300
- 2018Newsroom: A Dataset of 1.3 Million Summaries with Diverse Extractive Strategiescited by 292
- 2019Deep Semi-Supervised Anomaly Detectioncited by 290
- 2022Masked Autoencoders that Listencited by 271
- 2022OpenOOD: Benchmarking Generalized Out-of-Distribution Detectioncited by 267
- 2021SSD: A Unified Framework for Self-Supervised Outlier Detectioncited by 264
- 2022Mitigating Neural Network Overconfidence with Logit Normalizationcited by 251
- 2016Dense Associative Memory for Pattern Recognitioncited by 230
- 2022Extremely Simple Activation Shaping for Out-of-Distribution Detectioncited by 212
- 2022Out-of-Distribution Detection and Selective Generation for Conditional Language Modelscited by 193
- 2021DICE: Leveraging Sparsification for Out-of-Distribution Detectioncited by 166
- 2021PixMix: Dreamlike Pictures Comprehensively Improve Safety Measurescited by 131
- 2023Non-Parametric Outlier Synthesiscited by 129
- 2022How to Exploit Hyperspherical Embeddings for Out-of-Distribution Detection?cited by 128
- 2024To Believe or Not to Believe Your LLMcited by 120
- 2022POEM: Out-of-Distribution Detection with Posterior Samplingcited by 108
- 2018Abstractive Summarization of Reddit Posts with Multi-level Memory Networkscited by 102
- 2021Shifts: A Dataset of Real Distributional Shift Across Multiple Large-Scale Taskscited by 100
50 most cited of 79 All 79 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.