CLJan 12, 2024

Large Language Models Can Learn Temporal Reasoning

arXiv:2401.06853v6180 citationsh-index: 38ACL
Originality Incremental advance
AI Analysis

This work addresses a specific limitation in LLMs for temporal reasoning, representing an incremental advancement in enhancing their reasoning capabilities.

The paper tackles the challenge of temporal reasoning in large language models by introducing TG-LLM, a framework that uses temporal graphs and a synthetic dataset for fine-tuning, resulting in improved performance on TR tasks and benchmarks.

While large language models (LLMs) have demonstrated remarkable reasoning capabilities, they are not without their flaws and inaccuracies. Recent studies have introduced various methods to mitigate these limitations. Temporal reasoning (TR), in particular, presents a significant challenge for LLMs due to its reliance on diverse temporal concepts and intricate temporal logic. In this paper, we propose TG-LLM, a novel framework towards language-based TR. Instead of reasoning over the original context, we adopt a latent representation, temporal graph (TG) that enhances the learning of TR. A synthetic dataset (TGQA), which is fully controllable and requires minimal supervision, is constructed for fine-tuning LLMs on this text-to-TG translation task. We confirmed in experiments that the capability of TG translation learned on our dataset can be transferred to other TR tasks and benchmarks. On top of that, we teach LLM to perform deliberate reasoning over the TGs via Chain-of-Thought (CoT) bootstrapping and graph data augmentation. We observed that those strategies, which maintain a balance between usefulness and diversity, bring more reliable CoTs and final results than the vanilla CoT distillation.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes