← Search

Jiyan He

4 accepted papers

2026

One Tool Is Enough: Reinforcement Learning of LLM Agents for Repository-Level Code Navigation

ICML 2026poster

Locating files and functions requiring modification in large software repositories is challenging due to their scale and structural complexity. Existing LLM-based methods typically treat this as a repository-level retrieval task and rely on multiple auxiliary tools, which often overlook code executi…

Cited by 0SourceScholar
2024

AquaLoRA: Toward White-box Protection for Customized Stable Diffusion Models via Watermark LoRA

ICML 2024poster

Diffusion models have achieved remarkable success in generating high-quality images. Recently, the open-source models represented by Stable Diffusion (SD) are thriving and are accessible for customization, giving rise to a vibrant community of creators and enthusiasts. However, the widespread availa…

2024

Physical Consistency Bridges Heterogeneous Data in Molecular Multi-Task Learning

NeurIPS 2024poster

In recent years, machine learning has demonstrated impressive capability in handling molecular science tasks. To support various molecular properties at scale, machine learning models are trained in the multi-task learning paradigm. Nevertheless, data of different molecular properties are often not…

Cited by 1SourcePDFScholar
2023

Exploring the Limits of Differentially Private Deep Learning with Group-wise Clipping

ICLR 2023poster

Differentially private deep learning has recently witnessed advances in computational efficiency and privacy-utility trade-off. We explore whether further improvements along the two axes are possible and provide affirmative answers leveraging two instantiations of \emph{group-wise clipping}. To red…

Cited by 52SourcePDFScholar