SPD: Single Pass Decoding for Generative Reranking
2026-09-07 12:00Models🔥 42.2 heat score
1sources
1days unfolding
42.2heat score
1mentions
SummaryAI generated
The research team proposed a single-step decoding strategy called SPD (Single Forward Pass), aimed at addressing the issue of slow inference speed in generative reordering tasks. This strategy utilizes a lightweight self-attention head to directly extract the score matrix from the pre-filled hidden states of large language models (LLMs), and employs the Hungarian algorithm to directly calculate the optimal binary matching, thereby outputting N ordinal numbers in a single forward pass with an O(1) time complexity. Experiments show that this method, combined with LoRA fine-tuning and autoregressive distillation training, improves the inference speed from traditional methods to 28 milliseconds (a 64-fold increase), while maintaining the same ranking quality as the teacher model. Complete ablation experiments further verified the specific contributions of architecture, training signals, and backbone adaptation to performance. This framework successfully combines generative ranking with combinatorial optimization, providing a new technical approach for constructing other O(1) decoding mechanisms to support real-time ranking.