시각 언어 모델의 수리 능력 향상: CogAlign의 등장
본 기사는 시각 언어 모델(VLMs)의 시각적 산술 능력 부족 문제를 해결하기 위해 제안된 CogAlign이라는 새로운 후속 훈련 전략에 대해 다룹니다. 피아제의 인지 발달 이론에서 영감을 받은 CogAlign은 제한된 데이터로도 VLMs의 성능을 향상시키는 효과를 보이며, 향후 VLMs 연구에 중요한 전환점을 제시합니다.

시각 언어 모델의 수리적 한계 극복: CogAlign의 혁신
최근 괄목할 만한 발전을 이룬 시각 언어 모델(VLMs)은 다양한 다중 모드 작업에서 뛰어난 성능을 보여주고 있습니다. 하지만, 사물 개수 세기나 길이 비교와 같이 간단해 보이는 시각적 산술 능력에서는 어려움을 겪는 경우가 많습니다. 이러한 능력은 차트 이해나 기하학적 추론과 같은 복잡한 작업에 필수적입니다.
Kung-Hsiang Huang 등 연구진은 이러한 VLMs의 한계를 극복하기 위해, 기본적인 시각적 산술 능력에 초점을 맞춘 일련의 조사 작업을 통해 그 원인을 규명했습니다. 분석 결과, 사전 훈련된 시각 인코더는 충분한 정보를 포착하지만, 텍스트 디코더가 산술적 추론을 위해 이를 올바르게 디코딩하는 데 실패하는 경우가 많다는 것을 발견했습니다.
연구진은 이 문제를 해결하기 위해 피아제의 인지 발달 이론에서 영감을 얻은 새로운 후속 훈련 전략인 CogAlign을 제안했습니다. CogAlign은 시각적 변환 하에서 불변 속성을 인식하도록 VLMs를 훈련시키는 방법입니다. 실험 결과, CogAlign은 세 가지 다양한 VLMs의 성능을 크게 향상시켰습니다. 특히, CHOCOLATE 데이터셋에서는 평균 4.6%, MATH-VISION 데이터셋에서는 2.9%의 성능 향상을 보였으며, 지도 학습 방식보다 우수하거나 동등한 성능을 보였습니다. 또한, CogAlign은 기존 방식보다 훈련 데이터를 60% 적게 사용하면서도 동일한 효과를 달성했습니다.
이러한 결과는 CogAlign이 기본적인 시각적 산술 능력 향상과 하위 작업으로의 전이에 효과적이고 일반화 가능함을 보여줍니다. CogAlign은 VLMs의 한계를 극복하고, 더욱 발전된 인공지능 시스템 구축에 중요한 역할을 할 것으로 기대됩니다. 향후 연구에서는 CogAlign을 다양한 시각적 추론 작업에 적용하고, 그 효과를 더욱 심도 있게 분석하는 것이 필요할 것입니다.
요약: VLMs의 시각적 산술 능력 부족 문제를 해결하기 위해, 피아제 이론에 기반한 CogAlign이라는 혁신적인 후속 훈련 전략이 개발되었습니다. CogAlign은 제한된 데이터로도 VLMs의 성능을 향상시키며, 다양한 하위 작업에서의 성능 향상을 보여줍니다. 이는 향후 VLMs의 발전에 중요한 기여를 할 것으로 예상됩니다.
Reference
[arxiv] Why Vision Language Models Struggle with Visual Arithmetic? Towards Enhanced Chart and Geometry Understanding
Published: (Updated: )
Author: Kung-Hsiang Huang, Can Qin, Haoyi Qiu, Philippe Laban, Shafiq Joty, Caiming Xiong, Chien-Sheng Wu
http://arxiv.org/abs/2502.11492v1