Papers by Kaiyi Zhang
StepHint: Multi-level Stepwise Hints Enhance Reinforcement Learning to Reason (2026.acl-long)
Copied to clipboard
| Challenge: | Reinforcement learning with verifiable rewards (RLVR) approaches face two challenges: the near-miss reward problem and exploration stagnation. |
| Approach: | They propose an algorithm that partitions valid reasoning chains into reasoning steps using multi-level stepwise hints. |
| Outcome: | The proposed method outperforms competing RLVR enhancement methods across six mathematical benchmarks and two out-of-domain benchmarks. |
An Analysis and Mitigation of the Reversal Curse (2024.emnlp-main)
Copied to clipboard
| Challenge: | Recent research observes a phenomenon in large language models called the "reversal curse" when dealing with two entities, LLMs excel in handling sequences in the form of "aRb" but when asked "who is Mary Lee Pfeiffer's son?" the LLM exhibits considerable confusion and fails to provide a as the answer . |
| Approach: | They conduct the first-ever study of how the reversal curse happens in large language models . they find that LLMs excel in handling sequences in the form of "aRb" but struggle to provide a satisfactory answer when asked "who is Mary Lee Pfeiffer's son?" |
| Outcome: | The proposed study shows that the reversal curse can stem from specific training objectives . the study also shows that a reverse query can be difficult to understand . |
Batch-ICL: Effective, Efficient, and Order-Agnostic In-Context Learning (2024.findings-acl)
Copied to clipboard
| Challenge: | In-context learning (ICL) is a meta-optimization process that affects performance . we develop a batch-based inference algorithm that is order-agnostic to ICL examples . |
| Approach: | They develop an order-agnostic inference algorithm that aggregates ICL examples in batches . they find it outperforms most permutations of ICL, and it even exceeds the best order . |
| Outcome: | The proposed method outperforms standard ICL examples while reducing computational resources. |