Image-Text-to-Text
Transformers
Safetensors
fusion_phi
text-generation
conversational
custom_code
starriver030515 commited on
Commit
0fa09e7
·
verified ·
1 Parent(s): ae85a4a

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - starriver030515/FUSION-Pretrain-10M
4
+ - starriver030515/FUSION-Finetune-12M
5
+ base_model:
6
+ - microsoft/Phi-3.5-mini-instruct
7
+ - google/siglip-so400m-patch14-384
8
+ license: apache-2.0
9
+ ---
10
+ # Model Card for FUSION
11
+
12
+ This is the checkpoint after Stage 1, Stage1.5 and Stage2 training of FUSION-Phi3.5-3B.
13
+
14
+ ## Model Details
15
+
16
+ **Model Description**
17
+
18
+ <img src="https://raw.githubusercontent.com/starriver030515/FUSION/main/images/encoder.jpg" alt="encoder" width="1000px">
19
+
20
+ <img src="https://raw.githubusercontent.com/starriver030515/FUSION/main/images/decoder.jpg" alt="decoder" width="1000px">
21
+
22
+ FUSION is a family of multimodal large language models that adopts a fully integrated vision-language architecture, enabling comprehensive and fine-grained cross-modal understanding. In contrast to prior approaches that primarily perform shallow or late-stage modality fusion during the LLM decoding phase, FUSION achieves deep, dynamic integration across the entire vision-language processing pipeline.
23
+
24
+ To enable this, FUSION utilizes Text-Guided Unified Vision Encoding, which incorporates textual context directly into the vision encoder. This design allows for pixel-level vision-language alignment and facilitates early-stage cross-modal interaction.
25
+
26
+ During decoding, FUSION employs Context-Aware Recursive Alignment Decoding strategy. This component dynamically aggregates and refines visual features based on the evolving textual context at each decoding step, allowing the model to capture question-level semantics with high precision.
27
+
28
+ To further enhance alignment and reduce the semantic gap between modalities, FUSION integrates Dual-Supervised Semantic Mapping Loss, which provides simultaneous supervision in both visual and textual embedding spaces. This dual-path guidance strengthens the consistency and semantic coherence of the fused representations.
29
+
30
+ **Base Model**
31
+
32
+ **LLM**: [microsoft/Phi-3.5-mini-instruct](https://huggingface.co/microsoft/Phi-3.5-mini-instruct)
33
+
34
+ **Vision Encoder**: [google/siglip-so400m-patch14-384](https://huggingface.co/google/siglip-so400m-patch14-384)
35
+
36
+ ## Training Details
37
+
38
+ **Training Strategies**
39
+
40
+ FUSION is trained with a three-stage training framework, ensuring comprehensive alignment and integration between visual and linguistic modalities.
41
+
42
+ - **Stage1: Foundational Semantic Alignment**: We pretrain the vision encoder using extensive image-caption datasets to establish precise semantic alignment be- tween visual and textual representations.
43
+ - **Stage1.5: Contextual Multimodal Fusion**: In contrast to Stage 1, this intermediate stage incorporates various types of QA data along with image-caption pairs. This phase is designed to enhance the model’s adaptability in aligning vision and language representations across a broad spectrum of scenarios.
44
+ - **Stage2: Visual Instruction Tuning**: At this stage, we expose the model to various visual tasks, enabling it to answer downstream vision-related questions effectively.
45
+
46
+ **Training Data**
47
+
48
+ - [10M FUSION Alignment Data](https://huggingface.co/datasets/starriver030515/FUSION-Pretrain-10M) For Stage1
49
+ - [12M FUSION Curated Instruction Tuning Data](https://huggingface.co/datasets/starriver030515/FUSION-Finetune-12M) For Stage1.5 and Stage2
50
+
51
+ ## Performance
52
+
53
+ <img src="https://raw.githubusercontent.com/starriver030515/FUSION/main/images/performance.jpg" alt="performance" width="1000px">
54
+
55
+ **Where to send questions or comments about the model:**
56
+
57
+ https://github.com/starriver030515/FUSION/issues
58
+
59
+ ## Paper or resources for more information
60
+
61
+ - https://github.com/starriver030515/FUSION
62
+ - Coming soon~