Just Pass Twice: Efficient Token Classification with LLMs for Zero-Shot NER
Not on the map. Why?
Abstract (arXiv)
Large language models encode extensive world knowledge valuable for zero-shot named entity recognition. However, their causal attention mechanism, where tokens attend only to preceding context, prevents effective token classification when disambiguation requires future context. Existing approaches use LLMs generatively, prompting them to list entities or produce structured outputs, but suffer from slow autoregressive decoding, hallucinated entities, and formatting errors. We propose Just Pass Twice (JPT), a simple yet effective method that enables causal LLMs to perform discriminative token classification with full bidirectional context. Our key insight is that concatenating the input to itself lets each token in the second pass attend to the complete sentence, requiring no architectural modifications. We combine these representations with definition-guided entity embeddings for flexible zero-shot generalization. Our approach achieves state-of-the-art results on zero-shot NER benchmarks, surpassing the previous best method by +7.9 F1 on average across CrossNER and MIT benchmarks, being over 20x faster than comparable generative methods. Interactive demo and access to the model are available at https://witness-ai-jpt-ner.hf.space/
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 35 references that are in the map. First posted on arXiv 6 Apr 2026.
Builds on
21 of 35 references.
References · 21
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
- 2023Splitwise: Efficient generative LLM inference using phase splittingcited by 558
- 2022Efficient Training of Language Models to Fill in the Middlecited by 255
- 2023ChatIE: Zero-Shot Information Extraction via Chatting with ChatGPTcited by 250
- 2023A Comprehensive Capability Analysis of GPT-3 and GPT-3.5 Series Modelscited by 246
- 2015Named Entity Recognition with Bidirectional LSTM-CNNscited by 201
- 2023InstructUIE: Multi-task Instruction Tuning for Unified Information Extractioncited by 170
- 2023Evaluating ChatGPT's Information Extraction Capabilities: An Assessment of Performance, Explainability, Calibration, and Faithfulnesscited by 130
9 more
- 2023GoLLIE: Annotation Guidelines improve Zero-Shot Information-Extractioncited by 119
- 2019A Stack-Propagation Framework with Token-Level Intent Detection for Spoken Language Understandingcited by 107
- 2020CrossNER: Evaluating Cross-Domain Named Entity Recognitioncited by 104
- 2024NuNER: Entity Recognition Encoder Pre-training via LLM-Annotated Datacited by 34
- 2024TELEClass: Taxonomy Enrichment and LLM-Enhanced Hierarchical Text Classification with Minimal Supervisioncited by 25
- 2024Recent Advances in Named Entity Recognition: A Comprehensive Survey and Comparative Studycited by 20
- 2024Improving Minimum Bayes Risk Decoding with Multi-Promptcited by 13
- 2025A Systematic Characterization of LLM Inference on GPUscited by 8
- 2024Multi-Prompting Decoder Helps Better Language Understandingcited by 2
Built on by
Not cited yet.