Semantic Reranking at Inference Time for Hard Examples in Rhetorical Role Labeling
Not on the map. Why?
Abstract (arXiv)
Rhetorical Role Labeling (RRL) assigns a functional role to each sentence in a document and is widely used in legal, medical, and scientific domains. While language models (LMs) achieve strong average performance, they remain unreliable on hard examples, where prediction confidence is low. Existing approaches typically handle uncertainty implicitly and treat labels as discrete identifiers, overlooking the semantic information encoded in label names. We introduce RISE, an inference-time semantic reranking framework that leverages label semantics to refine predictions on hard instances. RISE automatically identifies low-confidence predictions and reranks model outputs using contrastively learned label representations, without retraining or modifying the underlying model. Experiments on eight domain-specific RRL datasets with seven LMs, including encoder-based and causal architectures, show an average gain of +9.15 macro-F1 points on hard examples. For explainability, we further propose manual hardness annotations to study difficulty from both model and human perspectives, revealing a moderate agreement with Cohen's kappa = 0.40.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 25 of its 39 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
25 of 39 references.
References · 25
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2024A Survey on LLM-as-a-Judgecited by 1,384
- 2024SaulLM-7B: A pioneering Large Language Model for Lawcited by 136
- 2017PubMed 200k RCT: a Dataset for Sequential Sentence Classification in Medical Abstractscited by 123
- 2024Uncertainty in Language Models: Assessment through Rank-Calibrationcited by 40
- 2022Corpus for Automatic Structuring of Legal Documentscited by 25
- 2024MAQA: Evaluating Uncertainty Quantification in LLMs Regarding Data Uncertaintycited by 25
13 more
- 2025When AIs Judge AIs: The Rise of Agent-as-a-Judge Evaluation for LLMscited by 16
- 2024HILL: Hierarchy-aware Information Lossless Contrastive Learning for Hierarchical Text Classificationcited by 13
- 2021Cross-Domain Multi-Task Learning for Sequential Sentence Classification in Research Paperscited by 11
- 2023Generation-driven Contrastive Self-training for Zero-shot Text Classification with Instruction-following LLMcited by 11
- 2023Harnessing GPT-3.5-turbo for Rhetorical Role Prediction in Legal Casescited by 8
- 2024Incubating Text Classifiers Following User Instruction with Nothing but LLMcited by 6
- 2024HiCuLR: Hierarchical Curriculum Learning for Rhetorical Role Labeling of Legal Documentscited by 6
- 2024Dynamic Label Name Refinement for Few-Shot Dialogue Intent Classificationcited by 6
- 2024Mind Your Neighbours: Leveraging Analogous Instances for Rhetorical Role Labeling for Legal Documentscited by 5
- 2025Evaluating Design Decisions for Dual Encoder-based Entity Disambiguationcited by 5
- 2025Fine-Tuning Causal LLMs for Text Classification: Embedding-Based vs. Instruction-Based Approachescited by 5
- 2023Joint Span Segmentation and Rhetorical Role Labeling with Data Augmentation for Legal Documentscited by 3
- 2025Label-semantics Aware Generative Approach for Domain-Agnostic Multilabel Classificationcited by 2
Built on by
Not cited yet.