Text Generation
Transformers
Safetensors
English
qwen3_5_moe
image-text-to-text
agent
deep-research
reasoning
tool-use
long-context
qwen3.5
mixture-of-experts
conversational
Instructions to use cfli/A-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use cfli/A-base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="cfli/A-base") messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] pipe(text=messages)# Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("cfli/A-base") model = AutoModelForMultimodalLM.from_pretrained("cfli/A-base", device_map="auto") messages = [ { "role": "user", "content": [ {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"}, {"type": "text", "text": "What animal is on the candy?"} ] }, ] inputs = processor.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(processor.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use cfli/A-base with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "cfli/A-base" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "cfli/A-base", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/cfli/A-base
- SGLang
How to use cfli/A-base with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "cfli/A-base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "cfli/A-base", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "cfli/A-base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "cfli/A-base", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use cfli/A-base with Docker Model Runner:
docker model run hf.co/cfli/A-base
Add files using upload-large-folder tool
Browse files- README.md +12 -34
- assets/arex-benchmark-results.svg +0 -0
README.md
CHANGED
|
@@ -16,16 +16,13 @@ tags:
|
|
| 16 |
---
|
| 17 |
|
| 18 |
<div align="center">
|
| 19 |
-
<img src="assets/arex-logo.png" width="
|
| 20 |
-
<
|
| 21 |
-
<
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
[](https://arex-research.com/)
|
| 27 |
-
[](https://huggingface.co/BAAI/AREX-Turbo)
|
| 28 |
-
|
| 29 |
</div>
|
| 30 |
|
| 31 |
## Introduction
|
|
@@ -40,22 +37,22 @@ This repository contains **AREX-Base**, the larger model in the AREX family. It
|
|
| 40 |
| :---: | :---: | :---: | :---: |
|
| 41 |
| Mixture-of-Experts | 122B total / 10B activated | 262,144 tokens | Up to 300 research turns and 5 self-improvement rounds |
|
| 42 |
|
| 43 |
-
|
| 44 |
|
| 45 |
- **Recursive self-improvement:** evaluates provisional answers and chooses whether to accept, refine, or restart the research trajectory.
|
| 46 |
- **Verification-guided research:** converts unresolved answer constraints into focused follow-up searches.
|
| 47 |
- **Autonomous context updates:** retains verified findings, source identifiers, rejected candidates, open constraints, and the next research plan.
|
| 48 |
-
- **Long-horizon tool use:**
|
| 49 |
- **Evidence-grounded answers:** maintains supporting evidence and answer-level confidence throughout the research process.
|
| 50 |
|
| 51 |
-
|
| 52 |
|
| 53 |
| Model | Architecture | Backbone | Context | Positioning |
|
| 54 |
| --- | --- | --- | ---: | --- |
|
| 55 |
| [AREX-Base](https://huggingface.co/BAAI/AREX-Base) | 122B total / 10B activated MoE | Qwen3.5-122B-A10B | 256K | Higher-capacity model with the strongest overall AREX results. |
|
| 56 |
| [AREX-Turbo](https://huggingface.co/BAAI/AREX-Turbo) | 4B dense | Qwen3.5-4B | 256K | Compact model for lower-cost research and tool-use settings. |
|
| 57 |
|
| 58 |
-
##
|
| 59 |
|
| 60 |
AREX organizes deep research as two interacting loops supported by autonomous context management.
|
| 61 |
|
|
@@ -65,32 +62,17 @@ AREX organizes deep research as two interacting loops supported by autonomous co
|
|
| 65 |
|
| 66 |
This design preserves useful work across iterations and reduces repeated exploration of paths that have already been rejected.
|
| 67 |
|
| 68 |
-
## Training
|
| 69 |
-
|
| 70 |
-
The AREX training pipeline combines:
|
| 71 |
-
|
| 72 |
-
- verified synthetic research tasks that require multi-hop evidence acquisition and constraint checking
|
| 73 |
-
- high-quality teacher trajectories filtered for coherent research behavior, valid tool use, evidence support, and answer confidence
|
| 74 |
-
- agentic mid-training for browse-intensive tool use, evidence acquisition, expert reasoning, and hypothesis verification
|
| 75 |
-
- long-horizon reinforcement learning with outcome rewards and step-level shaping at decision-critical research moments
|
| 76 |
-
|
| 77 |
-
Step-level rewards target actions such as finding decisive evidence, rejecting an incorrect path, redirecting a search, and updating context after meaningful progress.
|
| 78 |
-
|
| 79 |
## Evaluation
|
| 80 |
|
| 81 |
AREX is evaluated through a unified long-horizon search-agent interface with `search`, `visit`, `update_context`, and `finish` tools. HLE with tools additionally provides a Python tool. The reported setup allows up to 300 inner-loop turns and 5 outer-loop self-improvement operations.
|
| 82 |
|
| 83 |
-
|
| 84 |
-
<img src="assets/arex-benchmark-results.png" width="100%" alt="AREX benchmark results across six deep research benchmarks" />
|
| 85 |
-
</div>
|
| 86 |
|
| 87 |
AREX-Base scores **82.5** on BrowseComp, **85.4** on GAIA, **71.0** on xbench-2510, **89.9** on DeepSearchQA, **82.0** on WideSearch-en, and **52.4** on HLE with tools. It leads AREX-Turbo across all six reported benchmarks.
|
| 88 |
|
| 89 |
<details>
|
| 90 |
<summary><strong>Full benchmark table from the AREX paper</strong></summary>
|
| 91 |
|
| 92 |
-
<br>
|
| 93 |
-
|
| 94 |
**Frontier models**
|
| 95 |
|
| 96 |
| Model | BrowseComp | GAIA | xbench-2510 | DeepSearchQA | WideSearch-en | HLE (tool) |
|
|
@@ -130,7 +112,3 @@ DeepSearchQA is reported with Item-F1. The other benchmarks are reported with ac
|
|
| 130 |
## Intended Use
|
| 131 |
|
| 132 |
AREX-Base is intended for research-agent applications that require long-horizon information seeking, evidence aggregation, multi-constraint verification, and tool-augmented reasoning. Its larger capacity is suited to settings where research quality is prioritized over serving cost.
|
| 133 |
-
|
| 134 |
-
## Limitations
|
| 135 |
-
|
| 136 |
-
AREX depends on the availability and quality of its external search and browsing tools. It can still produce incorrect, incomplete, or weakly supported conclusions. Benchmark results also depend on the agent interface, tool implementation, inference configuration, and research budget described above. Outputs should be independently verified before high-stakes use.
|
|
|
|
| 16 |
---
|
| 17 |
|
| 18 |
<div align="center">
|
| 19 |
+
<img src="assets/arex-logo.png" width="48%" alt="AREX"><br>
|
| 20 |
+
<strong>AREX-Base</strong><br>
|
| 21 |
+
<sub>Towards a Recursively Self-Improving Agent for Deep Research</sub><br>
|
| 22 |
+
<a href="#"><img src="https://img.shields.io/badge/Paper-AREX-B31B1B?style=flat-square&logo=arxiv&logoColor=white" alt="Paper"></a>
|
| 23 |
+
<a href="https://545999961.github.io/AREX/web"><img src="https://img.shields.io/badge/GitHub-AREX-24292F?style=flat-square&logo=github&logoColor=white" alt="GitHub"></a>
|
| 24 |
+
<a href="https://arex-research.com/"><img src="https://img.shields.io/badge/Live_Demo-AREX-0F766E?style=flat-square&logo=googlechrome&logoColor=white" alt="AREX Demo"></a>
|
| 25 |
+
<a href="https://huggingface.co/BAAI/AREX-Turbo"><img src="https://img.shields.io/badge/Model-AREX--Turbo-FFD21E?style=flat-square&logo=huggingface&logoColor=000" alt="AREX-Turbo"></a>
|
|
|
|
|
|
|
|
|
|
| 26 |
</div>
|
| 27 |
|
| 28 |
## Introduction
|
|
|
|
| 37 |
| :---: | :---: | :---: | :---: |
|
| 38 |
| Mixture-of-Experts | 122B total / 10B activated | 262,144 tokens | Up to 300 research turns and 5 self-improvement rounds |
|
| 39 |
|
| 40 |
+
**Key Features**
|
| 41 |
|
| 42 |
- **Recursive self-improvement:** evaluates provisional answers and chooses whether to accept, refine, or restart the research trajectory.
|
| 43 |
- **Verification-guided research:** converts unresolved answer constraints into focused follow-up searches.
|
| 44 |
- **Autonomous context updates:** retains verified findings, source identifiers, rejected candidates, open constraints, and the next research plan.
|
| 45 |
+
- **Long-horizon tool use:** supports multi-round search, browsing, evidence integration, and answer construction.
|
| 46 |
- **Evidence-grounded answers:** maintains supporting evidence and answer-level confidence throughout the research process.
|
| 47 |
|
| 48 |
+
**Model Family**
|
| 49 |
|
| 50 |
| Model | Architecture | Backbone | Context | Positioning |
|
| 51 |
| --- | --- | --- | ---: | --- |
|
| 52 |
| [AREX-Base](https://huggingface.co/BAAI/AREX-Base) | 122B total / 10B activated MoE | Qwen3.5-122B-A10B | 256K | Higher-capacity model with the strongest overall AREX results. |
|
| 53 |
| [AREX-Turbo](https://huggingface.co/BAAI/AREX-Turbo) | 4B dense | Qwen3.5-4B | 256K | Compact model for lower-cost research and tool-use settings. |
|
| 54 |
|
| 55 |
+
## Method
|
| 56 |
|
| 57 |
AREX organizes deep research as two interacting loops supported by autonomous context management.
|
| 58 |
|
|
|
|
| 62 |
|
| 63 |
This design preserves useful work across iterations and reduces repeated exploration of paths that have already been rejected.
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
## Evaluation
|
| 66 |
|
| 67 |
AREX is evaluated through a unified long-horizon search-agent interface with `search`, `visit`, `update_context`, and `finish` tools. HLE with tools additionally provides a Python tool. The reported setup allows up to 300 inner-loop turns and 5 outer-loop self-improvement operations.
|
| 68 |
|
| 69 |
+

|
|
|
|
|
|
|
| 70 |
|
| 71 |
AREX-Base scores **82.5** on BrowseComp, **85.4** on GAIA, **71.0** on xbench-2510, **89.9** on DeepSearchQA, **82.0** on WideSearch-en, and **52.4** on HLE with tools. It leads AREX-Turbo across all six reported benchmarks.
|
| 72 |
|
| 73 |
<details>
|
| 74 |
<summary><strong>Full benchmark table from the AREX paper</strong></summary>
|
| 75 |
|
|
|
|
|
|
|
| 76 |
**Frontier models**
|
| 77 |
|
| 78 |
| Model | BrowseComp | GAIA | xbench-2510 | DeepSearchQA | WideSearch-en | HLE (tool) |
|
|
|
|
| 112 |
## Intended Use
|
| 113 |
|
| 114 |
AREX-Base is intended for research-agent applications that require long-horizon information seeking, evidence aggregation, multi-constraint verification, and tool-augmented reasoning. Its larger capacity is suited to settings where research quality is prioritized over serving cost.
|
|
|
|
|
|
|
|
|
|
|
|
assets/arex-benchmark-results.svg
ADDED
|
|