S*: 코드 생성을 위한 테스트 시간 확장의 혁신
본 기사는 S* 프레임워크를 소개하며, 기존의 코드 생성 방식의 한계를 뛰어넘는 혁신적인 성능 향상을 보여주는 연구 결과를 다룹니다. S*는 병렬 및 순차적 확장, 차별화된 입력 생성 및 실행 기반 정보 활용 등의 기술을 통해 다양한 LLM 및 LRM에서 일관된 성능 향상을 달성하며, 코드 생성 분야의 새로운 가능성을 제시합니다.

꿈꿔왔던 코드 생성의 미래, S*가 열다
최근 급격한 발전을 이룬 인공지능(AI) 분야에서, 특히 대규모 언어 모델(LLM)의 발전은 눈부십니다. 수학 분야에서 이미 그 효과가 입증된 테스트 시간 계산 능력 향상은 코드 생성 분야에서는 아직 미개척 영역으로 남아있었습니다. 하지만 이제, S 프레임워크*가 그 한계를 뛰어넘는 혁신을 가져왔습니다.
리 다청(Dacheng Li) 등 연구진이 발표한 논문 "S*: Test Time Scaling for Code Generation"에 따르면, S*는 기존의 병렬 확장 패러다임에 순차적 확장을 더한 하이브리드 접근 방식을 채택했습니다. 단순히 처리 속도만 높이는 것이 아니라, 새로운 선택 메커니즘을 통해 쌍방 비교를 위한 차별화된 입력을 생성하고, 실행 기반 정보를 활용하여 정확한 솔루션을 강력하게 식별합니다.
놀라운 성능 향상: 한계를 넘어선 결과
연구진은 12개의 LLM과 LRM(Large Reasoning Model)을 사용하여 S*의 성능을 평가했습니다. 그 결과는 놀라웠습니다.
- 모델 종류와 크기에 관계없이 일관된 성능 향상: 30억 파라미터 모델이 GPT-4o-mini를 능가하는 성과를 보였습니다. 이는 모델의 크기가 성능을 결정하는 절대적인 요소가 아님을 보여줍니다.
- 비추론 모델의 역전: S를 적용한 GPT-4o-mini는 추론 모델인 o1-preview를 LiveCodeBench에서 3.7%p 앞섰습니다. 이는 추론 능력이 코드 생성의 필수 요소가 아님을 시사하며, S가 모델의 본질적인 한계를 극복하는 데 도움을 줄 수 있음을 보여줍니다.
- 최첨단 추론 모델의 성능 향상: DeepSeek-R1-Distill-Qwen-32B에 S*를 적용한 결과, LiveCodeBench에서 85.7%의 정확도를 달성하여 o1 (high)의 88.5%에 근접했습니다.
이러한 결과는 S가 코드 생성 분야의 패러다임을 바꿀 잠재력을 가지고 있음을 명확히 보여줍니다. 더욱 놀라운 점은, Github (https://github.com/NovaSky-AI/SkyThought) 에서 코드를 공개하여 누구나 S를 활용할 수 있도록 했다는 것입니다. 이를 통해 AI 기반 코드 생성 기술의 대중화가 더욱 가속화될 것으로 예상됩니다.
앞으로의 전망: 더욱 발전된 코드 생성 기술을 기대하며
S의 등장은 단순히 기술적인 진보를 넘어, 더욱 효율적이고 정확한 코드 생성 시스템 구축에 대한 새로운 가능성을 제시합니다. 앞으로 S가 더욱 발전하여 더욱 복잡하고 다양한 코드 생성 작업에도 적용될 수 있기를 기대하며, 이를 통해 개발자들의 생산성 향상과 소프트웨어 개발의 혁신을 이끌어낼 수 있기를 기대합니다. 이는 단순한 기술의 발전이 아닌, 보다 나은 미래를 향한 중요한 한 걸음입니다.
Reference
[arxiv] S*: Test Time Scaling for Code Generation
Published: (Updated: )
Author: Dacheng Li, Shiyi Cao, Chengkun Cao, Xiuyu Li, Shangyin Tan, Kurt Keutzer, Jiarong Xing, Joseph E. Gonzalez, Ion Stoica
http://arxiv.org/abs/2502.14382v1