Momentum Posterior Regularization for Multi-hop Dense Retrieval

Zehua Xia, Yuyang Wu, Yiyun Xia, Cam Tu Nguyen


Abstract
Multi-hop question answering (QA) often requires sequential retrieval (multi-hop retrieval), where each hop retrieves missing knowledge based on information from previous hops. To facilitate more effective retrieval, we aim to distill knowledge from a posterior retrieval, which has access to posterior information like an answer, into a prior retrieval used during inference when such information is unavailable. Unfortunately, current methods for knowledge distillation in one-time retrieval are ineffective for multi-hop QA due to two issues: 1) posterior information is often defined as the response (i.e. answers), which may not clearly connect to the query without intermediate retrieval; and 2) the large knowledge gap between prior and posterior retrievals makes distillation using existing methods unstable, even resulting in performance loss. As such, we propose MoPo (Momentum Posterior Regularization) with two key innovations: 1) Posterior information of one hop is defined as a query-focus summary from the golden knowledge of the previous and current hops; 2) We develop an effective training strategy where the posterior retrieval is updated along with the prior retrieval via momentum moving average method, allowing smoother and effective distillation. Experiments on HotpotQA and StrategyQA demonstrate that MoPo outperforms existing baselines in both retrieval and downstream QA tasks.
Anthology ID:
2025.coling-main.550
Volume:
Proceedings of the 31st International Conference on Computational Linguistics
Month:
January
Year:
2025
Address:
Abu Dhabi, UAE
Editors:
Owen Rambow, Leo Wanner, Marianna Apidianaki, Hend Al-Khalifa, Barbara Di Eugenio, Steven Schockaert
Venue:
COLING
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
8255–8271
Language:
URL:
https://aclanthology.org/2025.coling-main.550/
DOI:
Bibkey:
Cite (ACL):
Zehua Xia, Yuyang Wu, Yiyun Xia, and Cam Tu Nguyen. 2025. Momentum Posterior Regularization for Multi-hop Dense Retrieval. In Proceedings of the 31st International Conference on Computational Linguistics, pages 8255–8271, Abu Dhabi, UAE. Association for Computational Linguistics.
Cite (Informal):
Momentum Posterior Regularization for Multi-hop Dense Retrieval (Xia et al., COLING 2025)
Copy Citation:
PDF:
https://aclanthology.org/2025.coling-main.550.pdf