Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

OODTE: A Differential Testing Engine for the ONNX Optimizer

Created by
  • Haebom

저자

Nikolaos Louloudakis, Ajitha Rajan

개요

본 논문은 ONNX Optimizer의 정확성을 자동으로 포괄적으로 평가하기 위한 유틸리티인 OODTE를 제시한다. OODTE는 차등 테스트 및 평가 방법론을 채택하여 ONNX 모델에 최적화를 적용하고 원본과 최적화된 버전을 모두 실행하여 최적화 중 발생하는 문제를 자동으로 캡처한다. 정확도 차이가 발생하면 OODTE는 더 세분화된 단계에서 프로세스를 반복하여 책임 있는 최적화 단계를 반복적으로 격리한다. 130개의 ONNX 모델을 사용한 평가 결과, 9.2%의 모델 인스턴스에서 최적화기가 충돌하거나 잘못된 모델이 생성되었으며, 분류 모델의 30%, 객체 탐지 및 분할 모델의 16.6%에서 원본과 최적화된 버전 간에 다른 출력이 나타났다. 텍스트 관련 작업에 중점을 둔 모델은 일반적으로 최적화에 강건했다. OODTE는 47개의 최적화 패스 중 9개와 최적화기 전반에 영향을 미치는 15개의 문제(이전에 알려지지 않은 14개 포함)를 발견했으며, 모든 문제는 ONNX Optimizer 팀에 보고되었다. OODTE는 ONNX 생태계를 넘어 AI 모델 최적화기를 검증하는 간단하지만 효과적인 프레임워크를 제공한다.

시사점, 한계점

시사점:
ONNX Optimizer의 정확성 문제를 체계적으로 검증하고, 기존에 알려지지 않은 여러 버그를 발견하여 ONNX Optimizer 개선에 기여.
OODTE는 다른 컴파일러 최적화기에도 적용 가능한 일반적인 프레임워크 제공.
다양한 모델 유형(분류, 객체 탐지, 분할, 텍스트 관련 작업 등)에 대한 광범위한 평가 수행.
한계점:
OODTE의 평가는 사용자 정의 입력 집합에 의존적이며, 입력 집합의 선택에 따라 평가 결과가 달라질 수 있음.
테스트된 모델의 수(130개)가 ONNX 모델 전체를 대표하지 못할 가능성 존재.
최적화 과정에서 발생하는 성능 저하에 대한 분석은 포함되지 않음.
👍