7 papers
Confidence-Calibrated Inference Expansion for Evaluator-Guided Test-Time Reasoning
Weida Liang, Kenji Kawaguchi
Evaluator-guided reasoning presents a local control problem at every step: trust the current answer, return a fallback, or spend more inference. We formulate this choice as a compu…
Bayesian Test-Time Inference of Task-Aligned Similarity from Weak Interactive Feedback
Weida Liang, Kenji Kawaguchi
Language model systems repeatedly make test-time decisions by comparing similarity: they retrieve exemplars, rank candidates, and revise responses. Yet similarity is not unique: le…
Lost in Adaptation: Layer-Selective Recovery of Temporal Reasoning in Video-Language Models
Zihang Fu, Haonan Wang, Jian Kang +2
Multimodal adaptation can weaken temporal reasoning (TR) in video-language models (VLMs) even when the visual evidence needed for an answer is correctly perceived. This weakened co…
Think in Parallel, Answer as One: Logit Averaging for Open-Ended Reasoning
Haonan Wang, Chao Du, Kenji Kawaguchi +1
Majority voting has proven effective for close-ended question answering by aggregating parallel reasoning traces. However, it is not directly applicable to open-ended reasoning, su…
From Harm to Help: Turning Reasoning In-Context Demos into Assets for Reasoning LMs
Haonan Wang, Weida Liang, Zihang Fu +8
Recent reasoning LLMs (RLMs), especially those trained with verifier-based reinforcement learning, often perform worse with few-shot CoT than with direct answering. We revisit this…
LoReUn: Data Itself Implicitly Provides Cues to Improve Machine Unlearning
Xiang Li, Qianli Shen, Haonan Wang +1
Recent generative models face significant risks of producing harmful content, which has underscored the importance of machine unlearning (MU) as a critical technique for eliminatin…
PromptArmor: Simple yet Effective Prompt Injection Defenses
Tianneng Shi, Kaijie Zhu, Zhun Wang +13
Despite their potential, recent research has demonstrated that LLM agents are vulnerable to prompt injection attacks, where malicious prompts are injected into the agent's input, c…