← Search

Jia Gu

4 accepted papers

2026

RLKD: Distilling LLMs’ Reasoning via Reinforcement Learning

AAAI 2026technical

Distilling reasoning paths from teacher to student models via supervised fine-tuning (SFT) provides a shortcut for improving the reasoning ability of the smaller Large Language Models (LLMs). However, the reasoning paths generated by teacher models often reflect only surface-level traces of their un

Cited by 0SourcePDFScholar
2026

Rethinking Bias in Generative Data Augmentation for Medical AI: A Frequency Recalibration Method

AAAI 2026technical

Developing Medical AI relies on large datasets and easily suffers from data scarcity. Generative data augmentation (GDA) using AI generative models offers a solution to synthesize realistic medical images. However, the bias in GDA is often underestimated in medical domains, with concerns about the r

Cited by 0SourcePDFScholar
2025

Do LLMs Play Dice? Exploring Probability Distribution Sampling in Large Language Models for Behavioral Simulation

COLING 2025main

With the rapid advancement of large language models (LLMs) for handling complex language tasks, an increasing number of studies are employing LLMs as agents to emulate the sequential decision-making processes of humans often represented as Markov decision-making processes (MDPs). The actions in MDPs…

Cited by 2SourcePDFScholar
2023

FAST: a Fused and Accurate Shrinkage Tree for Heterogeneous Treatment Effects Estimation

NeurIPS 2023poster

This paper proposes a novel strategy for estimating the heterogeneous treatment effect called the Fused and Accurate Shrinkage Tree ($\mathrm{FAST}$). Our approach utilizes both trial and observational data to improve the accuracy and robustness of the estimator. Inspired by the concept of shrinkag…

Cited by 1SourcePDFScholar