소프트웨어 이슈 해결의 혁신: SoRFT의 등장
Ma Zexiong 등 연구진이 개발한 SoRFT는 하위 작업 중심의 강화 학습 기법을 통해 LLM의 소프트웨어 이슈 해결 능력을 향상시키는 혁신적인 방법입니다. 오픈소스 모델 기반으로 비용 효율성을 확보하고, SWE-Bench 데이터셋을 통한 실험 결과 SOTA 성능을 달성했습니다.

최근 소프트웨어 개발에서 이슈 해결은 필수적인 과정입니다. 하지만 기존의 상용 모델 기반 프레임워크는 높은 비용과 개인 정보 보호 문제를 야기하며, 일반화 성능이 떨어지고 오픈소스 자원을 충분히 활용하지 못하는 한계를 지니고 있습니다.
이러한 문제점을 해결하기 위해 Ma Zexiong 등 연구진이 제시한 혁신적인 솔루션이 바로 SoRFT (Subtask-oriented Reinforced Fine-Tuning) 입니다. SoRFT는 대규모 언어 모델(LLM)의 이슈 해결 능력을 향상시키는 새로운 학습 방식입니다. 핵심은 이슈 해결 과정을 파일 위치 확인, 함수 위치 확인, 라인 위치 확인, 코드 수정 생성 등의 구조화된 하위 작업으로 분해하는 데 있습니다.
SoRFT는 크게 두 단계로 구성됩니다. 첫 번째 단계는 거절 샘플링을 통한 지도 학습 미세 조정입니다. Chain of Thought (CoT) 데이터를 정답을 기반으로 필터링하여 LLM을 미세 조정합니다. 두 번째 단계는 규칙 기반 강화 학습으로, 정답 기반 보상을 사용하는 PPO (Proximal Policy Optimization) 알고리즘을 활용합니다.
연구진은 SoRFT로 학습된 모델을 SWE-Bench Verified와 SWE-Bench Lite 데이터셋으로 평가했습니다. 그 결과, 오픈소스 모델 중 최고 성능(SOTA)을 달성하였습니다. 예를 들어, SoRFT-Qwen-7B 모델은 SWE-Bench Verified에서 21.4%의 이슈를 해결했습니다. 이는 SoRFT가 이슈 해결 성능을 크게 향상시키고 모델의 일반화 능력을 개선하며 상용 모델에 대한 비용 효율적인 대안을 제공함을 보여줍니다.
결론적으로, SoRFT는 하위 작업 중심의 강화 학습을 통해 LLM의 이슈 해결 능력을 획기적으로 향상시키는 혁신적인 기술입니다. 이는 오픈소스 기반의 소프트웨어 개발 생태계 발전에 크게 기여할 것으로 기대됩니다. 앞으로 SoRFT의 발전과 실제 적용 사례에 대한 지속적인 관심이 필요합니다. 특히, 다양한 프로그래밍 언어와 개발 환경에 대한 적용성 확장 연구가 중요한 다음 과제가 될 것입니다.
Reference
[arxiv] SoRFT: Issue Resolving with Subtask-oriented Reinforced Fine-Tuning
Published: (Updated: )
Author: Zexiong Ma, Chao Peng, Pengfei Gao, Xiangxin Meng, Yanzhen Zou, Bing Xie
http://arxiv.org/abs/2502.20127v1