Validated Code Translation for Projects with External Libraries

1. 문제 정의 (Problem)

대형 언어 모델 (LLM) 은 Go 와 같은 시스템 코드를 메모리 안전한 언어인 Rust 로 번역하는 데 유망한 도구로 부상했습니다. 그러나 **외부 라이브러리 (External Libraries)**에 의존하는 프로젝트를 번역할 때 기존 접근 방식은 다음과 같은 중대한 한계에 직면합니다.

1. 할루시네이션 및 API 매핑 실패: LLM 은 타겟 언어 (Rust) 에 존재하지 않는 라이브러리나 API 를 생성하거나 (할루시네이션), 기존 API 를 잘못 사용하는 경우가 많습니다. 특히 Rust 는 Go 와 달리 크레이트 (crate) 와 모듈 구조가 복잡하여, 올바른 API 를 선택하는 것뿐만 아니라 해당 API 를 호출 가능하게 만드는 적절한 import 문을 생성하는 것이 매우 어렵습니다.
2. 검증의 어려움 (Opaque Types): 외부 라이브러리가 정의한 타입 (예: 암호화 키, 내부 필드가 숨겨진 타입) 은 불투명 (opaque) 합니다. 기존 검증 기술은 입력/출력 타입의 구조적 동일성을 가정하거나 필드 단위 접근을 전제로 하는데, 불투명 타입은 이러한 접근이 불가능합니다. 따라서 번역된 코드가 원본과 의미적으로 동등한지 (Semantic Equivalence) 검증하는 것이 매우 어렵습니다.

2. 제안된 방법론 (Methodology)

저자들은 Go 프로젝트를 Rust 로 번역하고 이를 검증하기 위해 **검색 증강 생성 (RAG)**과 교차 언어 검증 파이프라인을 결합한 프레임워크를 제안합니다.

2.1. 크레이트 범위 지식 베이스 및 RAG (Cross-Language Library Mapping)

LLM 의 할루시네이션을 줄이고 정확한 API 매핑을 위해 다음과 같은 단계를 거칩니다.

• 크레이트 매핑: Go 패키지를 대응되는 Rust 크레이트로 매핑합니다.
• 지식 베이스 구축: 각 Rust 크레이트의 공개 API, 문서, 그리고 호출을 가능하게 하는 유효한 import 경로를 포함하는 구조화된 지식 베이스를 사전에 구축합니다.
• 검색 및 재순위화: Go 함수의 문서와 시그니처를 기반으로 관련 Rust API 후보를 검색 (Dense Retrieval) 하고, LLM 을 통해 재순위화하여 가장 적합한 API 와 필요한 import 문을 프롬프트에 제공합니다.

2.2. 번역 생성 (Translation Generation)

• LLM 은 위 RAG 를 통해 제공된 컨텍스트 (API 시그니처, 문서, import 경로) 를 바탕으로 Go 코드를 Rust 로 번역합니다.
• 번역은 모듈 단위로 수행되며, 의존성 요약 정보를 바탕으로 각 함수와 타입을 독립적으로 번역합니다.

2.3. 불투명 타입을 위한 교차 언어 검증 (Cross-Language Validation)

번역된 코드가 컴파일되는 것을 넘어 기능적 동등성을 검증하기 위해 **공통 중개 데이터 형식 (Intermediate Data Format, IDF)**을 사용합니다.

• Protobuf 기반 캐리어 (Carrier) 생성: Go 와 Rust 간의 값 교환을 위해 Protobuf 스키마 (TProto) 를 생성합니다. 이는 언어 중립적인 포맷으로, Go 와 Rust 모두에서 직렬화/역직렬화 가능한 '캐리어 타입'을 생성합니다.
• 어댑터 합성 (Adapter Synthesis):
  • 원본 Go 타입 ↔ Protobuf 캐리어
  • 번역된 Rust 타입 ↔ Protobuf 캐리어
  • 이 변환은 라이브러리의 공개 API 만을 사용하여 합성됩니다. 불투명 타입의 내부 필드에 직접 접근하는 대신, 라이브러리가 제공하는 생성자나 직렬화 루틴을 사용하여 변환 코드를 생성합니다.
• 라운드트립 검증 (Round-trip Validation): 생성된 어댑터가 정보를 잃지 않고 원본 값을 복원할 수 있는지 (Go → Proto → Go) 검증합니다.
• 차등 I/O 검증 (Differential I/O Validation): 검증된 어댑터를 사용하여 동일한 단위 테스트 입력을 Go 와 Rust 함수에 각각 전달하고, 출력 결과를 비교하여 동등성을 확인합니다.

3. 주요 기여 (Key Contributions)

1. 외부 라이브러리 의존성 해결: LLM 기반 번역에서 외부 라이브러리 API 매핑 실패와 import 오류를 해결하기 위해 RAG 기반의 크레이트 매핑 및 import 해결 메커니즘을 처음 도입했습니다.
2. 불투명 타입을 위한 검증 프레임워크: 구조적으로 다른 불투명 타입을 가진 함수 간의 의미적 동등성을 검증할 수 있는 새로운 방법을 제시했습니다. 이는 필드 단위 접근이 불가능한 라이브러리 타입을 처리하기 위해 공개 API 만을 사용하는 어댑터 합성과 Protobuf 기반 캐리어를 활용합니다.
3. 실제 프로젝트 적용: Go 에서 Rust 로의 전체 저장소 번역을 목표로 하며, 실제 복잡한 외부 의존성을 가진 6 개의 Go 저장소를 대상으로 평가했습니다.

4. 실험 결과 (Results)

저자들은 6 개의 실제 Go 저장소 (암호화, 키 파생, 시간 파싱 등 다양한 도메인 포함) 를 대상으로 실험을 수행했습니다.

• 성공률 향상:
  • 제안된 방법 (CrossCrate) 은 모든 벤치마크에서 컴파일 성공률 95.83%, **동등성 검증 성공률 95.83%**를 기록했습니다.
  • 외부 의존성이 가장 많은 경우 (age 프로젝트) 에는 100% 성공률을 달성했습니다.
  • 기존 방법 (Oxidizer 등) 에 비해 평균적으로 약 2 배 이상의 성공률 향상을 보였습니다.
• Ablation Study (성분 분석):
  • RAG(지식 베이스) 와 Import 해결 기능을 모두 제거한 경우, 외부 라이브러리를 사용하는 함수의 컴파일 및 동등성 성공률은 **0%**로 떨어졌습니다.
  • 이는 올바른 API 매핑과 호출 가능한 import 문 생성이 번역 성공에 필수적임을 입증했습니다.

5. 의의 및 결론 (Significance)

이 논문은 LLM 기반 코드 번역이 외부 라이브러리에 의존하는 현실적인 프로젝트에서도 적용 가능함을 입증했습니다.

• 기술적 혁신: 불투명 타입과 복잡한 라이브러리 의존성을 가진 코드의 번역 및 검증을 가능하게 하는 어댑터 합성과 공통 캐리어 아키텍처는 향후 다른 언어 간 번역 (예: C++ to Rust, Java to Kotlin 등) 에도 적용 가능한 중요한 패러다임입니다.
• 실용성: 메모리 안전성을 위해 레거시 시스템을 Rust 로 마이그레이션하려는 기업 및 개발자들에게, 외부 라이브러리 의존성으로 인한 실패 위험을 줄이고 검증된 번역을 제공할 수 있는 강력한 도구를 제시합니다.

요약하자면, 이 연구는 RAG 를 통한 정확한 API 매핑과 Protobuf 기반의 교차 언어 검증 파이프라인을 결합하여, 외부 라이브러리가 포함된 복잡한 Go 프로젝트를 Rust 로 안전하고 정확하게 번역하는 새로운 표준을 제시했습니다.