[NeurIPS 2026] InfiniteVL: Synergizing Linear and Sparse Attention for Highly-Efficient, Unlimited-Input Vision-Language Models
Python
118
15 commits
updated Sep 24, 2026
Hongyuan Tao1, Bencheng Liao1, Shaoyu Chen2, Haoran Yin2, Qian Zhang2, Wenyu Liu1, Xinggang Wang1,✉️
1Huazhong University of Science and Technology, 2Horizon Robotics
(✉️) corresponding author: xgwang@hust.edu.cn
InfiniteVL is an efficient Vision-Language Model that combines linear attention for compact long-term memory with sparse attention for precise visual perception. It achieves Transformer-level multimodal performance while supporting highly efficient long-context processing.
Based on InfiniteVL, we develop two specialized variants:
Sep. 25th, 2026: InfiniteVL has been accepted by NeurIPS 2026! 🎉🎉🎉Aug. 28th, 2026: 🔥 Today's released GLM-5.3 and Qwen3.8-next both adopt linear + sparse attention as their core architecture — the design InfiniteVL explored and fully open-sourced (paper + training code + checkpoints) last year. If you want to understand, reproduce, or build on this architecture, this repo is a complete starting point!Feb. 2nd, 2026: 🚀 We have released the full training code and scripts! You can now reproduce our results following the Training Strategy.Dec. 10th, 2025: We release the InfiniteVL model weights and inference code! Please check Model Zoo.Dec. 10th, 2025: We release our paper on Arxiv.We recommend using Anaconda or Miniconda to manage the environment. The code is tested on Python 3.11 + PyTorch 2.6.0 + CUDA 12.1.
1. Clone the repository:
git clone https://github.com/hustvl/InfiniteVL.git
cd InfiniteVL
2. Create and activate a virtual environment:
conda create -n infinitevl python=3.11 -y
conda activate infinitevl
3. Install Environment:
pip install -r requirements.txt
InfiniteVL combines sparse attention for precise visual perception with linear attention for efficient long-term memory. Based on this architecture, we develop two variants for different long-context scenarios:
Our training follows a three-stage progressive knowledge-transfer pipeline:
Our training codebase is built upon LLaMA-Factory. Please refer to their repository for detailed documentation and framework usage.
We provide standard scripts to reproduce our three-stage training pipeline. The training is launched via train.sh.
Note: Before running, please modify the
model_name_or_path,datasetandcheckpoint_pathinside the.yamlfiles to point to your local directories.
1. Distillation Pretraining (layer to layer)
bash train.sh examples/train_linear/train_config_distill_stage1.yaml
2. Distillation Pretraining (End to End)
bash train.sh examples/train_linear/train_config_distill_stage2.yaml
3. Instruction SFT
bash train.sh examples/train_linear/train_config_distill_sft.yaml
InfiniteVL is engineered for unlimited-input scenarios. Unlike Transformer-based models where cost grows linearly with history, InfiniteVL maintains constant computational cost and memory usage.
Hardware Setup: All efficiency results are measured on a single NVIDIA RTX 4090 GPU.
InfiniteVL achieves state-of-the-art performance among linear-complexity VLMs. Crucially, thanks to our Hybrid Architecture and High-quality training strategies, it overcomes the traditional weakness of linear models in information-intensive tasks (e.g., OCR, Document Understanding), achieving results comparable to top-tier Transformer VLMs.
Key Takeaways:
We release two versions of InfiniteVL-4B to cater to different application scenarios.
| Model | Stage | Description | Training context Length | Download |
|---|---|---|---|---|
| InfiniteVL-4B | Stage 2 | Best Generalist / Base. The checkpoint directly after Instruction SFT. It delivers the peak foundational performance on standard multimodal benchmarks (e.g., OCR, MMMU, MathVista) and preserves the most robust knowledge. | 8K | 🤗 Hugging Face |
| InfiniteVL-4B-LongSFT | Stage 3 | Long-Context Adapted. Fine-tuned using only a small amount of long-sequence multimodal data. It successfully activates length generalization for streaming scenarios, though its full potential on extreme contexts is not yet fully exploited. | 32K | 🤗 Hugging Face |
💡 Recommendations:
- For Long-Context Inference: Please use the Stage 3 model. It enables stable streaming inference and avoids memory explosion.
- For Training / Fine-tuning: We strongly recommend using the Stage 2 model as your starting point. Since it maintains the strongest general capabilities and hasn't shifted towards the specific long-context distribution, it serves as the best foundation for adaptation to new tasks or domains.
Unlike Transformer-based VLMs where the KV cache grows dynamically, InfiniteVL maintains a constant-size memory state. This unique property allows us to use CUDA Graphs to capture the entire computation graph for both streaming prefill and decoding, eliminating kernel launch overheads and maximizing GPU utilization.
This is the key technology behind our 24 FPS real-time streaming performance.
Unlike Transformer-based VLMs where the KV cache grows dynamically, InfiniteVL maintains a constant-size memory state. This unique property allows us to use CUDA Graphs to capture the entire computation graph for streaming prefill, eliminating kernel launch overheads.
We provide a complete script in examples/demo_streaming_inference.py to demonstrate this capability.
🎥 Simulation Note: This script simulates a real-time streaming scenario by reading a local video file frame-by-frame. It treats the video as a continuous data stream, updating the global linear memory state on-the-fly without retraining.
⚠️ Requirement: This demo relies on the specialized model implementation (supporting
StaticCachePreallocand CUDA Graphs) located in theinfinitevl/infinitevl_streamingdirectory. Please ensure your environment is set up correctly to import these modules.
# Make sure you are in the project root
python examples/demo_streaming_inference.py \
--model_path /path/to/InfiniteVL-4B \
--video_path assets/demo.mp4 \
--fps 30
In addition to streaming prefill, InfiniteVL natively supports CUDA Graph-accelerated decoding. By capturing the decoding step into a static graph, we can achieve extremely low-latency token generation, further enhancing the responsiveness of real-time interactions.
🚧 Coming Soon: The code for accelerated decoding is currently being refactored and cleaned up. We are working hard to release it as soon as possible. Please stay tuned!
We provide visualization cases to demonstrate InfiniteVL's robust performance across diverse scenarios, ranging from information-intensive static tasks to ultra-long streaming video understanding.
InfiniteVL effectively overcomes the traditional limitations of linear attention in detailed visual perception. By combining Sliding Window Attention with Gated DeltaNet, it excels at Dense Text Recognition (OCR), Chart Interpretation, and Complex Scene Description, delivering performance comparable to full-attention Transformers.
The core strength of InfiniteVL lies in its ability to maintain coherent memory over unlimited input streams.
The examples below demonstrate a continuous street-view video stream. InfiniteVL maintains a constant memory state and accurately answers questions at various timestamps (e.g., Frame 3100, ~1M tokens processed), recalling specific details like "NBC Studios" text or the color of a pedestrian's bag without forgetting.
If you have any questions, please contact Hongyuan Tao via email (hongyuantao@hust.edu.cn).
If you find InfiniteVL useful for your research or applications, please consider citing our paper:
@article{tao2025infinitevl,
title={InfiniteVL: Synergizing Linear and Sparse Attention for Highly-Efficient, Unlimited-Input Vision-Language Models},
author={Tao, Hongyuan and Liao, Bencheng and Chen, Shaoyu and Yin, Haoran and Zhang, Qian and Liu, Wenyu and Wang, Xinggang},
journal={arXiv preprint},
year={2025}
}
InfiniteVL is built upon the giants of the open-source community. We would like to express our gratitude to:
216 followers · starred Mar 2026
Python
99.8%
[NeurIPS 2026] InfiniteVL: Synergizing Linear and Sparse Attention for Highly-Efficient, Unlimited-Input Vision-Language Models
Python
118
15 commits
updated Sep 24, 2026
Hongyuan Tao1, Bencheng Liao1, Shaoyu Chen2, Haoran Yin2, Qian Zhang2, Wenyu Liu1, Xinggang Wang1,✉️
1Huazhong University of Science and Technology, 2Horizon Robotics
(✉️) corresponding author: xgwang@hust.edu.cn
InfiniteVL is an efficient Vision-Language Model that combines linear attention for compact long-term memory with sparse attention for precise visual perception. It achieves Transformer-level multimodal performance while supporting highly efficient long-context processing.
Based on InfiniteVL, we develop two specialized variants:
Sep. 25th, 2026: InfiniteVL has been accepted by NeurIPS 2026! 🎉🎉🎉Aug. 28th, 2026: 🔥 Today's released GLM-5.3 and Qwen3.8-next both adopt linear + sparse attention as their core architecture — the design InfiniteVL explored and fully open-sourced (paper + training code + checkpoints) last year. If you want to understand, reproduce, or build on this architecture, this repo is a complete starting point!Feb. 2nd, 2026: 🚀 We have released the full training code and scripts! You can now reproduce our results following the Training Strategy.Dec. 10th, 2025: We release the InfiniteVL model weights and inference code! Please check Model Zoo.Dec. 10th, 2025: We release our paper on Arxiv.We recommend using Anaconda or Miniconda to manage the environment. The code is tested on Python 3.11 + PyTorch 2.6.0 + CUDA 12.1.
1. Clone the repository:
git clone https://github.com/hustvl/InfiniteVL.git
cd InfiniteVL
2. Create and activate a virtual environment:
conda create -n infinitevl python=3.11 -y
conda activate infinitevl
3. Install Environment:
pip install -r requirements.txt
InfiniteVL combines sparse attention for precise visual perception with linear attention for efficient long-term memory. Based on this architecture, we develop two variants for different long-context scenarios:
Our training follows a three-stage progressive knowledge-transfer pipeline:
Our training codebase is built upon LLaMA-Factory. Please refer to their repository for detailed documentation and framework usage.
We provide standard scripts to reproduce our three-stage training pipeline. The training is launched via train.sh.
Note: Before running, please modify the
model_name_or_path,datasetandcheckpoint_pathinside the.yamlfiles to point to your local directories.
1. Distillation Pretraining (layer to layer)
bash train.sh examples/train_linear/train_config_distill_stage1.yaml
2. Distillation Pretraining (End to End)
bash train.sh examples/train_linear/train_config_distill_stage2.yaml
3. Instruction SFT
bash train.sh examples/train_linear/train_config_distill_sft.yaml
InfiniteVL is engineered for unlimited-input scenarios. Unlike Transformer-based models where cost grows linearly with history, InfiniteVL maintains constant computational cost and memory usage.
Hardware Setup: All efficiency results are measured on a single NVIDIA RTX 4090 GPU.
InfiniteVL achieves state-of-the-art performance among linear-complexity VLMs. Crucially, thanks to our Hybrid Architecture and High-quality training strategies, it overcomes the traditional weakness of linear models in information-intensive tasks (e.g., OCR, Document Understanding), achieving results comparable to top-tier Transformer VLMs.
Key Takeaways:
We release two versions of InfiniteVL-4B to cater to different application scenarios.
| Model | Stage | Description | Training context Length | Download |
|---|---|---|---|---|
| InfiniteVL-4B | Stage 2 | Best Generalist / Base. The checkpoint directly after Instruction SFT. It delivers the peak foundational performance on standard multimodal benchmarks (e.g., OCR, MMMU, MathVista) and preserves the most robust knowledge. | 8K | 🤗 Hugging Face |
| InfiniteVL-4B-LongSFT | Stage 3 | Long-Context Adapted. Fine-tuned using only a small amount of long-sequence multimodal data. It successfully activates length generalization for streaming scenarios, though its full potential on extreme contexts is not yet fully exploited. | 32K | 🤗 Hugging Face |
💡 Recommendations:
- For Long-Context Inference: Please use the Stage 3 model. It enables stable streaming inference and avoids memory explosion.
- For Training / Fine-tuning: We strongly recommend using the Stage 2 model as your starting point. Since it maintains the strongest general capabilities and hasn't shifted towards the specific long-context distribution, it serves as the best foundation for adaptation to new tasks or domains.
Unlike Transformer-based VLMs where the KV cache grows dynamically, InfiniteVL maintains a constant-size memory state. This unique property allows us to use CUDA Graphs to capture the entire computation graph for both streaming prefill and decoding, eliminating kernel launch overheads and maximizing GPU utilization.
This is the key technology behind our 24 FPS real-time streaming performance.
Unlike Transformer-based VLMs where the KV cache grows dynamically, InfiniteVL maintains a constant-size memory state. This unique property allows us to use CUDA Graphs to capture the entire computation graph for streaming prefill, eliminating kernel launch overheads.
We provide a complete script in examples/demo_streaming_inference.py to demonstrate this capability.
🎥 Simulation Note: This script simulates a real-time streaming scenario by reading a local video file frame-by-frame. It treats the video as a continuous data stream, updating the global linear memory state on-the-fly without retraining.
⚠️ Requirement: This demo relies on the specialized model implementation (supporting
StaticCachePreallocand CUDA Graphs) located in theinfinitevl/infinitevl_streamingdirectory. Please ensure your environment is set up correctly to import these modules.
# Make sure you are in the project root
python examples/demo_streaming_inference.py \
--model_path /path/to/InfiniteVL-4B \
--video_path assets/demo.mp4 \
--fps 30
In addition to streaming prefill, InfiniteVL natively supports CUDA Graph-accelerated decoding. By capturing the decoding step into a static graph, we can achieve extremely low-latency token generation, further enhancing the responsiveness of real-time interactions.
🚧 Coming Soon: The code for accelerated decoding is currently being refactored and cleaned up. We are working hard to release it as soon as possible. Please stay tuned!
We provide visualization cases to demonstrate InfiniteVL's robust performance across diverse scenarios, ranging from information-intensive static tasks to ultra-long streaming video understanding.
InfiniteVL effectively overcomes the traditional limitations of linear attention in detailed visual perception. By combining Sliding Window Attention with Gated DeltaNet, it excels at Dense Text Recognition (OCR), Chart Interpretation, and Complex Scene Description, delivering performance comparable to full-attention Transformers.
The core strength of InfiniteVL lies in its ability to maintain coherent memory over unlimited input streams.
The examples below demonstrate a continuous street-view video stream. InfiniteVL maintains a constant memory state and accurately answers questions at various timestamps (e.g., Frame 3100, ~1M tokens processed), recalling specific details like "NBC Studios" text or the color of a pedestrian's bag without forgetting.
If you have any questions, please contact Hongyuan Tao via email (hongyuantao@hust.edu.cn).
If you find InfiniteVL useful for your research or applications, please consider citing our paper:
@article{tao2025infinitevl,
title={InfiniteVL: Synergizing Linear and Sparse Attention for Highly-Efficient, Unlimited-Input Vision-Language Models},
author={Tao, Hongyuan and Liao, Bencheng and Chen, Shaoyu and Yin, Haoran and Zhang, Qian and Liu, Wenyu and Wang, Xinggang},
journal={arXiv preprint},
year={2025}
}
InfiniteVL is built upon the giants of the open-source community. We would like to express our gratitude to:
216 followers · starred Mar 2026
Python
99.8%