Command Palette
Search for a command to run...
네이티브 멀티모달 사전 학습의 제로-베이스 확장
네이티브 멀티모달 사전 학습의 제로-베이스 확장
Haoyuan Wu Aoqi Wu Hai Wang Jiajia Wu Jinxiang Ou Bei Yu
초록
대규모 언어 모델(LLM)이 놀라운 추론 능력을 보여주지만, 텍스트 전용 사전 학습에 의존하기 때문에 멀티모달 물리 세계에 대한 인식이 제한된다. 네이티브 멀티모달 사전 학습은 멀티모달 입력에 대해 모델을 처음부터 학습시킴으로써 이러한 한계를 피하며, 이를 통해 심층적인 교차 모달 통합을 달성하고 기존의 후기 융합 아키텍처에 내재된 최적화 비대칭성을 완화한다. 이러한 장점에도 불구하고, 이 패러다임의 확장 특성은 체계적으로 규명되지 않은 상태로 남아 있다. 이 격차를 해소하기 위해, 우리는 고정된 계산 예산 하에서 트랜스포머 기반 비전-언어 모델을 학습시키기 위한 최적의 모델 크기와 토큰 수를 조사한다. 최소 목표 손실은 예측 가능한 계산 법칙을 따르는 반면, 계산 최적 모델 크기와 토큰 수는 멱법칙에 따라 확장됨을 입증한다. 특히, 언어 목표와 멀티모달 목표는 뚜렷이 다른 확장 행동을 보인다. 언어 할당 법칙은 데이터 구성에 거의 불변하여, 멀티모달 데이터 비율과 관계없이 안정적인 언어 학습이 이루어짐을 시사한다. 반대로, 멀티모달 할당 법칙은 이러한 구성에 매우 민감하다. 구체적으로, 텍스트 비중이 높은 혼합은 더 큰 모델 규모에서만 계산 효율적이 되어, 최적 자원 할당을 더 큰 모델 용량 쪽으로 이동시킨다. 또한, 데이터 구성이 계산 법칙과 할당 지수에 미치는 영향을 모델링함으로써, 모델 크기, 토큰 수, 데이터 혼합의 정확한 구성을 명시하는 효율성 프런티어를 도출한다. 다운스트림 평가는 네이티브 멀티모달 사전 학습이 긍정적인 교차 모달 전이를 유도하여 순수 텍스트 공간 추론을 향상시키고 강력한 멀티모달 문맥 내 학습을 가능하게 함을 추가로 보여준다. 요약하자면, 본 경험적 연구는 멀티모달 기반 모델을 예측 가능하게 확장하기 위한 필수적인 기반을 마련한다.
One-sentence Summary
Researchers from The Chinese University of Hong Kong and Tencent investigate scaling laws for native multimodal pre-training, finding that language and multimodal objectives exhibit distinct compute-optimal behaviors, with data composition strongly influencing multimodal allocation, and they derive an efficiency frontier for model size, token count, and data mixture, while also demonstrating that native pre-training improves pure-text spatial reasoning and multimodal in-context learning via positive cross-modal transfer.
Key Contributions
- Decoupling the joint training loss into separate language and multimodal objectives reveals that language scaling is largely invariant to data composition, while multimodal scaling is highly sensitive to the multimodal data ratio.
- A compute-optimal frontier is derived that explicitly accounts for data mixture, specifying precise allocations of model size and token count as functions of compute budget and data composition.
- Downstream experiments show that native multimodal pre-training induces positive cross-modal transfer, improving pure-text spatial reasoning and enabling robust multimodal in-context learning.
Introduction
While LLMs have strong reasoning, text-only training lacks multimodal grounding. Late-fusion methods alleviate this by coupling a pre-trained language model with a vision encoder, but they introduce a fundamental representation asymmetry because vision and language are learned separately. Native multimodal pre-training from scratch promises deeper integration, yet no scaling laws existed to guide how to allocate a fixed compute budget between model size and data across modalities. The authors fill this gap by deriving compute-optimal scaling laws for native multimodal pre-training, decoupling language and multimodal objectives. They find that language scaling follows composition-invariant allocation rules, while multimodal scaling is highly sensitive to the data mixture, leading to a joint Pareto frontier that prescribes the optimal model size, text token count, and multimodal token count for any compute budget. They also show that native pre-training transfers spatial reasoning to pure-text tasks and enables multimodal in-context learning.
Dataset
The authors construct a training corpus that mixes text and multimodal data:
- Text subset: 250B tokens sourced from web pages, books, academic papers, and similar domains.
- Multimodal subset: 75B tokens produced from web‑crawled image‑text pairs and interleaved image‑text documents. Images are converted into continuous patch embeddings using a single patch embedding layer, turning each image into a sequence of token-like embeddings. No additional vision encoder is used.
The entire dataset (text + multimodal tokens) is used to train a decoder‑only Mixture of Experts (MoE) Transformer. The patch embeddings are fed directly to the model alongside text tokens, treating images as native token sequences. No cropping, filtering, or metadata construction details are provided beyond the standard web‑scale curation.
Method
The authors revisit the fundamental problem of compute-optimal allocation in native multimodal pre-training. Given a fixed computational budget C, the goal is to optimally allocate resources between the model size N (the number of activated non-embedding parameters) and the total number of training tokens D. This trade-off is parameterized by the multimodal data ratio r. The final pre-training loss L(N,D) is minimized under the constraint C=6ND. Since language and multimodal objectives (Ltext and Lmm) are optimized simultaneously using shared parameters, the authors decouple the allocation problem. They define the effective compute for text and multimodal objectives as Ctext=6NDtext and Cmm=6NDmm, where Dtext=D/(1+r) and Dmm=Dr/(1+r).
To estimate the compute-optimal allocation, two independent methodologies are employed. The primary estimator is the IsoFLOP profile method. At a predetermined compute budget C, plotting the loss of each model against logN generates an IsoFLOP profile, which is modeled as a parabola. The minimum of this parabola identifies the optimal model size Nopt(C), from which the optimal token count follows as Dopt(C)=C/(6Nopt). As shown in the figure below:
The IsoFLOP curves for the language objective reveal stable parabolic minima across various values of r, demonstrating that an optimal model size exists for any given FLOP budget.
The second methodology is the training-curve envelope, which serves as an independent cross-validation mechanism. For a fixed model size N, increasing the token count D produces a trajectory of loss versus total compute C. Pooling the trajectories of all evaluated models and extracting the lower envelope yields the minimum achievable loss for any given budget C. Refer to the figure below:
This diagram illustrates the training curve envelope for the language objective. By extracting the envelope of minimal loss per FLOP from these curves across varying values of r and model sizes, the authors estimate the optimal model size and training token allocation. The compute frontier strictly follows a power law L(C)=E+(Cc/C)β.
Regarding the experimental setup, the model is trained on a mixture of 250B text tokens and 75B multimodal tokens. The architecture employs a Mixture of Experts (MoE) based on a decoder-only Transformer. Instead of traditional vision encoders, a single patch embedding layer projects images directly into continuous patch embeddings. The MoE models are trained using an auxiliary-loss-free approach.
The decoupled analysis reveals distinct scaling behaviors. The compute-optimal allocation for the language objective is highly composition-invariant. The IsoFLOP parametric fits and the training-curve envelope show that the parameter capacity required to minimize Ltext is strictly governed by the isolated budget Ctext, demonstrating empirical robustness to the introduction of multimodal tokens. In contrast, the multimodal objective exhibits strictly composition-variant scaling behavior. As r increases, the optimal model size exponent decreases substantially, confirming the inherently data-hungry nature of cross-modal alignment. Processing increasingly dense multimodal data shifts the optimal compute allocation heavily toward data scaling rather than parameter expansion.
To address practical native multimodal pre-training, which must optimize the shared parameter count N under a unified computational budget Ctotal, the authors establish a strict Pareto frontier. They employ an asymmetric modeling framework that pairs a composition-invariant language objective with a composition-variant multimodal objective. This asymmetry reflects the structural divergence between the modalities. Text-based language modeling relies on a robust statistical structure, while visual tokens provide external context. Consequently, scaling a unified multimodal foundation model requires a deliberate architectural shift, constraining theoretical parameter expansion in favor of training on substantially larger token budgets to accommodate the data requirements of dense multimodal inputs.
Experiment
The evaluation setup uses 16 text benchmarks and 23 multimodal benchmarks to assess base models under in-context learning. Native multimodal pre-training preserves core language capabilities, with no degradation in aggregate text performance, and produces cross-modal transfer where spatial reasoning from visual data improves purely text-based spatial tasks. Multimodal in-context learning emerges naturally with larger model scale and more training data, with the strongest few-shot gains concentrated on spatial reasoning benchmarks rather than recognition-focused tasks.