Conference Paper
Machine Learning

SelfCite

Yung-Sung Chuang, Benjamin Cohen-Wang, Shannon Zejiang Shen, Zhaofeng Wu, Hu Xu, Xi Victoria Lin, James Glass, Shang-Wen Li, Wen-tau Yih
February 1, 2025International Conference on Machine Learning18 citations

18

Citations

1

Influential Citations

International Conference on Machine Learning

Venue

2025

Year

Abstract

A self-supervised method for improving the quality of citations generated by LLMs. It uses context ablation to create a reward signal based on the necessity and sufficiency of cited text, guiding best-of-N sampling or preference optimization (like SimPO) to train LLMs to generate better citations without relying on human annotations.

Analysis

Why This Paper Matters

This paper addresses a critical challenge in large language model (LLM) deployment: generating accurate and well-attributed citations. As LLMs are increasingly used in knowledge-intensive tasks, the ability to cite sources correctly is essential for trustworthiness and verifiability. Traditional approaches rely on human-annotated data, which is costly and difficult to scale. SelfCite proposes a self-supervised alternative that leverages the model's own behavior to create a reward signal, making it a practical solution for improving citation quality at scale.

The significance lies in its potential to democratize high-quality citation generation. By removing the dependency on human annotations, this method can be applied to any LLM and domain, accelerating the adoption of LLMs in fields like journalism, academia, and legal analysis where citation accuracy is paramount.

Technical Contributions

  • Context Ablation Reward: The core innovation is a reward signal derived by comparing the model's output with and without the cited context. This measures how necessary and sufficient the citation is for the generated text.
  • Self-Supervised Training: The reward is used in a best-of-N sampling strategy to select high-quality citations, and also integrated into preference optimization (SimPO) to fine-tune the LLM directly.
  • No Human Annotations: The entire pipeline operates without human-labeled data, making it scalable and cost-effective.

Results

The paper reports improvements in citation quality metrics such as relevance and sufficiency compared to baseline LLMs. The self-supervised approach achieves competitive performance with methods that use human annotations, demonstrating its effectiveness. Specific metrics include higher precision and recall in citation attribution tasks.

Significance

SelfCite has broad implications for AI safety and reliability. By enabling self-supervised citation improvement, it paves the way for more trustworthy LLM outputs in real-world applications. This work also contributes to the growing field of self-supervised learning for alignment, showing that reward signals can be derived from the model's own behavior rather than external supervision. Future work could extend this approach to other aspects of output quality, such as factuality and coherence.