SCOUT: Selective Coupling via Optimal Unbalanced Transport for Interpretable Text Classification
Junhao Jia, Hanwen Zheng, Yueyi Wu, Huangwei Chen, Haishuai Wang, Jiajun Bu, Lei Wu
Abstract
Natural language data is inherently noisy, yet standard interpretable models often rely on scalar similarities that obscure the true evidentiary basis of a prediction. This limitation is particularly detrimental to prototype-based classification, where traditional full-alignment mechanisms force non-informative background segments to match informative prototypes, yielding unstable or misleading explanations. To mitigate this, we present SCOUT, a novel paradigm that grounds prototype reasoning in the selective correspondence of discriminative fragments. Concretely, we represent each document as a discrete distribution over span embeddings and employ differentiable Unbalanced Optimal Transport (UOT) to align them with class-specific prototypes. Unlike standard methods, this mechanism enables the model to focus strictly on decisive evidence while leaving irrelevant noise unmatched via geometric mass suppression. To ensure verifiability, we anchor prototype supports to readable training spans, establishing a transparent bridge between input segments and stored knowledge. Comprehensive experiments on seven benchmarks demonstrate that SCOUT yields prototypes focused on semantically significant spans, significantly outperforming traditional rationale extraction and post-hoc attribution methods in terms of faithfulness and stability.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on10
- ERASER: A Benchmark to Evaluate Rationalized NLP ModelsJay DeYoung, Sarthak Jain, Nazneen Fatema Rajani, Eric P. Lehman et al.ACL 2020 · 36 citations
- Toward Interpretable Semantic Textual Similarity via Optimal Transport-based Contrastive Sentence LearningSeonghyeon Lee, Dongha Lee, Seongbo Jang, Hwanjo YuACL 2022 · 25 citations
- Robust Representation Learning with Reliable Pseudo-labels Generation via Self-Adaptive Optimal Transport for Short Text ClusteringXiaolin Zheng, Mengling Hu, Weiming Liu, Chaochao Chen et al.ACL 2023 · 12 citations
- ProtoLens: Advancing Prototype Learning for Fine-Grained Interpretability in Text ClassificationBowen Wei, Ziwei ZhuACL 2025 · 7 citations
- MProto: Multi-Prototype Network with Denoised Optimal Transport for Distantly Supervised Named Entity RecognitionShuhui Wu, Yongliang Shen, Zeqi Tan, Wenqi Ren et al.EMNLP 2023 · 4 citations
Related papers
- Rationalizing Text Matching: Learning Sparse Alignments via Optimal TransportKyle Swanson, Lili Yu, Tao LeiACL 2020 · 3 citations
- Reducing Semantic Mismatch in Brain-to-Text Decoding Through Personalized Multimodal MaskingJiaxuan Chen, Yu Qi, Yueming Wang, Gang PanICLR 2026
- Explainable Legal Case Matching via Inverse Optimal Transport-based Rationale ExtractionWeijie Yu, Zhongxiang Sun, Jun Xu, Zhenhua Dong et al.SIGIR 2022 · 45 citations
- Optimal Transport for LLM Reward Modeling from Noisy FeedbackLicheng Pan, Haocheng Yang, Haoxuan Li, Yunsheng Lu et al.ICML 2026
- Noise-Aware Few-Shot Learning through Bi-directional Multi-View Prompt AlignmentLu Niu, Cheng XueCVPR 2026
