ALIGNBEAM : Inference-Time Alignment Transfer via Cross-Vocabulary Logit Mixing

Chirag Chawla, Pratinav Seth, Vinay Kumar Sankarapu


Abstract
Domain fine-tuning degrades the safety of large language models: fine-tuned specialists readily comply with harmful prompts framed in domain language. Existing inference-time defenses that mix logits from a safe anchor model require both models to share a vocabulary, which rules them out for the cross-family specialists where safety is most degraded. We present ALIGNBEAM, a training-free method that lifts this restriction by translating anchor probability mass into the target model’s vocabulary token-by-token at each decoding step; a small LLM judge then selects the safest among K candidate continuations. No weights are changed, and the mixing strength can be tuned at deployment without retraining. Across both cross-vocabulary and same-vocabulary evaluation pairs, ALIGNBEAM substantially raises refusal on structured, front-loaded adversarial benchmarks, with a 4.6-6.6 x inference slowdown inside the band of existing inference-time defenses. The results show that early-token refusal behavior can be transferred between model families at inference time, without touching either model’s weights.
Anthology ID:
2026.emnlp-main.2655
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
60438–60464
Language:
URL:
https://aclanthology.org/2026.emnlp-main.2655/
DOI:
Bibkey:
Cite (ACL):
Chirag Chawla, Pratinav Seth, and Vinay Kumar Sankarapu. 2026. ALIGNBEAM : Inference-Time Alignment Transfer via Cross-Vocabulary Logit Mixing. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 60438–60464, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
ALIGNBEAM : Inference-Time Alignment Transfer via Cross-Vocabulary Logit Mixing (Chawla et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.2655.pdf