Yihong Dong

Yihong Dong

Auto-generated profile, based on publicly available information

Yihong Dong

Peking University, Beijing, China

Research Focus & Metrics

Academic Career Length

6Years (2021-Present)

(Calculated once Scilit publication data is synced: current year − year of first publication + 1)

Representative Publications

View All Publications
27 September 2024 in ACM Transactions on Software Engineering and Methodology
Verified
Journal Article
Open access
YDYihong Dong
Yihong Dong
XJXue Jiang
Xue Jiang
ZJZhi Jin
Zhi Jin
GLGe Li
Ge Li
Although large language models (LLMs) have demonstrated remarkable code-generation ability, they still struggle with complex tasks. In real-world software development, humans usually tackle complex tasks through collaborative teamwork, a strategy that significantly controls development complexity and enhances software quality. Inspired by this, we present a self-collaboration framework for code generation employing LLMs, exemplified by ChatGPT. Specifically, through role instructions, (1) Multiple LLM agents act as distinct “experts,” each responsible for a specific subtask within a complex task; (2) Specify the way to collaborate and interact, so that different roles form a virtual team to facilitate each other’s work, ultimately the virtual team addresses code generation tasks collaboratively without the need for human intervention. To effectively organize and manage this virtual team, we incorporate software-development methodology into the framework. Thus, we assemble an elementary team consisting of three LLM roles (i.e., analyst, coder, and tester) responsible for software development’s analysis, coding, and testing stages. We conduct comprehensive experiments on various code-generation benchmarks. Experimental results indicate that self-collaboration code generation relatively improves 29.9–47.1% Pass@1 compared to the base LLM agent. Moreover, we showcase that self-collaboration could potentially enable LLMs to efficiently handle complex repository-level tasks that are not readily solved by the single LLM agent.

27 September 2024 in ACM Transactions on Software Engineering and Methodology
Verified
Journal Article
Open access
XJXue Jiang
Xue Jiang
YDYihong Dong
Yihong Dong
LWLecheng Wang
Lecheng Wang
ZFZheng Fang
Zheng Fang
QSQiwei Shang
Qiwei Shang
WJWenpin Jiao
Wenpin Jiao
Although large language models (LLMs) have demonstrated impressive ability in code generation, they are still struggling to address the complicated intent provided by humans. It is widely acknowledged that humans typically employ planning to decompose complex problems and schedule solution steps prior to implementation. To this end, we introduce planning into code generation to help the model understand complex intent and reduce the difficulty of problem-solving. This paper proposes a self-planning code generation approach with large language models, which consists of two phases, namely planning phase and implementation phase. Specifically, in the planning phase, LLM plans out concise solution steps from the intent combined with few-shot prompting. Subsequently, in the implementation phase, the model generates code step by step, guided by the preceding solution steps. We conduct extensive experiments on various code-generation benchmarks across multiple programming languages. Experimental results show that self-planning code generation achieves a relative improvement of up to 25.4% in Pass@1 compared to direct code generation, and up to 11.9% compared to Chain-of-Thought of code generation. Moreover, our self-planning approach also enhances the quality of the generated code with respect to correctness, readability, and robustness, as assessed by humans.

23 February 2025 in ACM Transactions on Software Engineering and Methodology
Verified
Journal Article
Open access
YDYihong Dong
Yihong Dong
JDJiazheng Ding
Jiazheng Ding
XJXue Jiang
Xue Jiang
GLGe Li
Ge Li
ZLZhuo Li
Zhuo Li
ZJZhi Jin
Zhi Jin
A proper code evaluation metric (CEM) profoundly impacts the evolution of code generation, which is an important research field in NLP and software engineering. Prevailing match-based CEMs (e.g., BLEU, Accuracy, and CodeBLEU) suffer from two significant drawbacks. 1. They primarily measure the surface differences between codes without considering their functional equivalence. However, functional equivalence is pivotal in evaluating the effectiveness of code generation, as different codes can perform identical operations. 2. They are predominantly designed for the Ref-only input format. However, code evaluation necessitates versatility in input formats. Aside from Ref-only, there are NL-only and Ref and NL formats, which existing match-based CEMs cannot effectively accommodate. In this article, we propose CodeScore, a large language model (LLM)-based CEM, which estimates the functional correctness of generated code on three input types. To acquire CodeScore, we present UniCE, a unified code generation learning framework, for LLMs to learn code execution (i.e., learning PassRatio and Executability of generated code) with unified input. Extensive experimental results on multiple code evaluation datasets demonstrate that CodeScore absolutely improves up to 58.87% correlation with functional correctness compared to other CEMs, achieves state-of-the-art performance, and effectively handles three input formats.

31 March 2021 in IEEE Transactions on Signal Processing
Verified
Journal Article
Closed access
YDYihong Dong
Yihong Dong
XJXiaohan Jiang
Xiaohan Jiang
Huaji ZhouHuaji Zhou
Huaji Zhou
Yun LinYun Lin
Yun Lin
QSQingjiang Shi
Qingjiang Shi
Signal recognition is one of the significant and challenging tasks in the signal processing and communications field. It is often a common situation that there's no training data accessible for some signal classes to perform a recognition task. Hence, as widely-used in image processing field, zero-shot learning (ZSL) is also very important for signal recognition. Unfortunately, ZSL regarding this field has hardly been studied due to inexplicable signal semantics. This paper proposes a ZSL framework, signal recognition and reconstruction convolutional neural networks (SR2CNN), to address relevant problems in this situation. The key idea behind SR2CNN is to learn the representation of signal semantic feature space by introducing a proper combination of cross entropy loss, center loss and reconstruction loss, as well as adopting a suitable distance metric space such that semantic features have greater minimal inter-class distance than maximal intra-class distance. The proposed SR2CNN can discriminate signals even if no training data is available for some signal class. Moreover, SR2CNN can gradually improve itself in the aid of signal detection, because of constantly refined class center vectors in semantic feature space. These merits are all verified by extensive experiments with ablation studies.

22 March 2021 in IEEE Transactions on Cognitive Communications and Networking
Verified
Journal Article
Closed access
YDYihong Dong
Yihong Dong
XJXiaohan Jiang
Xiaohan Jiang
LCLei Cheng
Lei Cheng
QSQingjiang Shi
Qingjiang Shi
Due to the emergence of deep learning, signal recognition has made great strides in performance improvement. The success of most deep learning methods relies on the accessibility of abundant labeled training data. However, the annotation of signals is quite expensive, making it challenging to train deep learning models substantially. This calls for the development of semi-supervised learning (SSL) method to fully utilize the unlabeled data to assist the training of deep learning models. To achieve this goal, three types of loss functions, tailored to the task of SLL-based signal recognition, are carefully designed in this paper. Together with a carefully selected neural network structure, the proposed SSL method can effectively extract the information from unlabeled training data and thus overcome the difficulty of insufficient training. Extensive numerical results using open source datasets are presented to show the superior performance of the proposed SSL method.

11 July 2026 in ACM Transactions on Software Engineering and Methodology
Verified
Journal Article
Open access
XJXue Jiang
Xue Jiang
YDYihong Dong
Yihong Dong
ZFZhiyuan Fan
Zhiyuan Fan
ZJZhi Jin
Zhi Jin
WJWenpin Jiao
Wenpin Jiao
GLGe Li
Ge Li
Although Large Language Models (LLMs) have made significant progress in code generation, they still struggle with code generation tasks in specific scenarios. These scenarios usually necessitate the adaptation of LLMs to fulfill specific needs, but the limited training data available in practice leads to poor code generation performance. Therefore, how to effectively adapt LLMs to new scenarios with few training data is a major challenge for current code generation. In this article, we propose a novel adaptation approach named Data-Efficient adaptation with Error-Driven Learning (DEED) for code generation. D EED leverages the errors made by LLMs as learning opportunities, using error revision to overcome their own shortcomings, thus achieving efficient learning. Specifically, D EED involves identifying error code generated by LLMs, employing S elf-Revise for code revision, optimizing the model with revised code, and iteratively adapting the process for continuous improvement. Experimental results show that, compared to other mainstream fine-tuning approaches, D EED achieves superior performance with few training data, showing an average relative improvement of 46.2% in Pass@1 on multiple code generation benchmarks. We also validate the effectiveness of S elf-Revise, which generates revised code that optimizes the model more efficiently compared to the code samples from datasets. Moreover, D EED consistently demonstrates strong performance across various LLMs, underscoring its applicability.