← Search

Hantian Ding

11 accepted papers

2026

Cyber-Zero: Training Cybersecurity Agents without Runtime

ICLR 2026poster

Large Language Models (LLMs) have achieved remarkable success in software engineering tasks when trained with executable runtime environments, particularly in resolving GitHub issues. However, such runtime environments are often unavailable in other domains, especially cybersecurity, where challenge…

Cited by 0SourcecodeScholar
2026

Training Language Model Agents to Find Vulnerabilities with CTF-Dojo

ICML 2026poster

Large language models (LLMs) have demonstrated exceptional capabilities when trained within executable runtime environments, notably excelling at software engineering tasks through verified feedback loops. Yet, scalable and generalizable execution-grounded environments remain scarce, limiting progre…

Cited by 0SourceScholar
2025

Planning-Aware Code Infilling via Horizon-Length Prediction

EMNLP 2025

Fill-in-the-Middle (FIM), or infilling, has become integral to code language models, enabling generation of missing code given both left and right contexts. However, the current FIM training paradigm which performs next-token prediction (NTP) over reordered sequence often leads to models struggling

Cited by 0SourcePDFScholar
2024

Bifurcated Attention for Single-Context Large-Batch Sampling

ICML 2024poster

In our study, we present bifurcated attention, a method developed for language model inference in single-context batch sampling contexts. This approach aims to reduce redundant memory IO costs, a significant factor in latency for high batch sizes and long context lengths. Bifurcated attention achiev…

Cited by 1SourcePDFScholar
2024

CODE REPRESENTATION LEARNING AT SCALE

ICLR 2024poster

Recent studies have shown that code language model at scale demonstrate significant performance gains on downstream tasks, i.e., code generation. However, most of the existing works on code representation learning train models at a hundred million parameter scale using very limited pretraining corpo…

Cited by 18SourcePDFScholar
2024

Fewer Truncations Improve Language Modeling

ICML 2024poster

In large language model training, input documents are typically concatenated together and then split into sequences of equal length to avoid padding tokens. Despite its efficiency, the concatenation approach compromises data integrity—it inevitably breaks many documents into incomplete pieces, leadi…

Cited by 20SourcePDFScholar
2023

A Static Evaluation of Code Completion by Large Language Models

ACL 2023industry

Large language models trained on code have shown great potential to increase productivity of software developers. Several execution-based benchmarks have been proposed to evaluate functional correctness of model-generated code on simple programming problems. Nevertheless, it is expensive to perform…

2023

CrossCodeEval: A Diverse and Multilingual Benchmark for Cross-File Code Completion

NeurIPS 2023poster

Code completion models have made significant progress in recent years, yet current popular evaluation datasets, such as HumanEval and MBPP, predominantly focus on code completion tasks within a single file. This over-simplified setting falls short of representing the real-world software development…

Cited by 122SourcePDFScholar
2023

Exploring Continual Learning for Code Generation Models

ACL 2023short

Large-scale code generation models such as Copilot and CodeT5 have achieved impressive performance. However, libraries are upgraded or deprecated very frequently and re-training large-scale language models is computationally expensive. Therefore, Continual Learning (CL) is an important aspect that r…

2023

Multi-lingual Evaluation of Code Generation Models

ICLR 2023top-25%

We present two new benchmarks, MBXP and Multilingual HumanEval, designed to evaluate code completion models in over 10 programming languages. These datasets are generated using a conversion framework that transpiles prompts and test cases from the original MBPP and HumanEval datasets into the corres…

2022

Towards Open-Domain Topic Classification

NAACL 2022system demonstrations

We introduce an open-domain topic classification system that accepts user-defined taxonomy in real time. Users will be able to classify a text snippet with respect to any candidate labels they want, and get instant response from our web interface. To obtain such flexibility, we build the backend mod…

Cited by 18SourcePDFScholar