hyx_194
commited on
Commit
·
ba814b7
1
Parent(s):
ea937d1
update captilization
Browse files
README.md
CHANGED
@@ -5,34 +5,32 @@ tags: []
|
|
5 |
|
6 |
# MERaLiON
|
7 |
|
8 |
-
MERaLiON-AudioLLM is a Speech-Text Large Language Model tailored for Singapore’s multilingual and multicultural landscape. Integrating a localised Whisper-large-v2 speech encoder and SEA-
|
9 |
-
of Singapore's local accents and dialects.
|
10 |
|
11 |
-
MERaLiON stands for <i
|
12 |
|
13 |
- **Developed by:** I<sup>2</sup>R, A\*STAR
|
14 |
- **Funded by:** Singapore NRF
|
15 |
- **Model type:** MultiModal LLM
|
16 |
- **Language(s) (Speech):** English (Global & Singapore)
|
17 |
- **Language(s) (NLP):** English, Chinese, Vietnamese, Indonesian, Thai, Filipino, Tamil, Malay, Khmer, Lao, Burmese, Javanese, Sundanese
|
18 |
-
- **License:**
|
19 |
|
20 |
For more details, please refer to our [report]().
|
21 |
|
22 |
## Model Description
|
23 |
|
24 |
-
MERaLiON-AudioLLM is designed to take in an audio-text pair as
|
25 |
-
input and generates a text output.
|
26 |
|
27 |
-
The architecture comprises three key components: an audio encoder that transforms speech or audio inputs into sequences of vector representations, a text decoder that interprets and responds to natural language instructions, and an adaptor module that compresses the encoder representations while aligning the encoder’s hidden dimension with the text decoder’s embedding size.
|
28 |
|
29 |
-
Specifically, we fine-tuned the MERaLiON-Whisper encoder from Whisper-large-v2 for the audio encoder and used SEA-LION V3, a localised LLM developed by our partner AI Singapore as the text decoder.
|
30 |
|
31 |
<img src="model_architecture.png" alt="model_architecture" width="400" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
|
32 |
|
33 |
## Capabilities
|
34 |
|
35 |
-
MERaLiON-AudioLLM is trained to address 8 tasks, including Automatic Speech Recognition (ASR), Speech Translation (ST), Spoken Question Answering (SQA), Spoken Dialogue Summarization (SDS), Speech Instruction (SI), Paralinguistics (PARA), Audio Captioning (AC), and Audio Scene Question Answering (ASQA).
|
36 |
|
37 |
[More information about the 8 tasks and evaluation results]
|
38 |
|
@@ -138,7 +136,7 @@ MERaLiON-AudioLLM is trained on a diverse collection of publicly available datas
|
|
138 |
|
139 |
### Compute and Infrastructure
|
140 |
|
141 |
-
MERaLiON-AudioLLM is trained on the ASPIRE 2A
|
142 |
|
143 |
With a global batch size of 640, we train the current release of MERaLiON-AudioLLM for around 200k steps, which took 2 days to complete using 16 nodes, 128 H100 GPUs.
|
144 |
|
|
|
5 |
|
6 |
# MERaLiON
|
7 |
|
8 |
+
MERaLiON-AudioLLM is a Speech-Text Large Language Model tailored for Singapore’s multilingual and multicultural landscape. Integrating a localised [Whisper-large-v2](https://huggingface.co/openai/whisper-large-v2) speech encoder and [SEA-LION V3](https://huggingface.co/aisingapore/gemma2-9b-cpt-sea-lionv3-instruct) text decoder, MERaLiON-AudioLLM is finetuned on **260,000 hours of speech and audio data**, **8 various tasks**, to address the diverse linguistic nuances of Singapore's local accents and dialects.
|
|
|
9 |
|
10 |
+
MERaLiON stands for <i>**M**ultimodal **E**mpathetic **R**easoning **a**nd **L**earning **i**n **O**ne **N**etwork</i>.
|
11 |
|
12 |
- **Developed by:** I<sup>2</sup>R, A\*STAR
|
13 |
- **Funded by:** Singapore NRF
|
14 |
- **Model type:** MultiModal LLM
|
15 |
- **Language(s) (Speech):** English (Global & Singapore)
|
16 |
- **Language(s) (NLP):** English, Chinese, Vietnamese, Indonesian, Thai, Filipino, Tamil, Malay, Khmer, Lao, Burmese, Javanese, Sundanese
|
17 |
+
- **License:** MIT
|
18 |
|
19 |
For more details, please refer to our [report]().
|
20 |
|
21 |
## Model Description
|
22 |
|
23 |
+
MERaLiON-AudioLLM is designed to take in an **audio-text pair** as input and generates a text output.
|
|
|
24 |
|
25 |
+
The architecture comprises three key components: an **audio encoder** that transforms speech or audio inputs into sequences of vector representations, a **text decoder** that interprets and responds to natural language instructions, and an **adaptor module** that compresses the encoder representations while aligning the encoder’s hidden dimension with the text decoder’s embedding size.
|
26 |
|
27 |
+
Specifically, we fine-tuned the **MERaLiON-Whisper** encoder from Whisper-large-v2 for the audio encoder and used SEA-LION V3, a localised LLM developed by our partner AI Singapore as the text decoder.
|
28 |
|
29 |
<img src="model_architecture.png" alt="model_architecture" width="400" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
|
30 |
|
31 |
## Capabilities
|
32 |
|
33 |
+
MERaLiON-AudioLLM is trained to address 8 tasks, including `Automatic Speech Recognition` (ASR), `Speech Translation` (ST), `Spoken Question Answering` (SQA), `Spoken Dialogue Summarization` (SDS), `Speech Instruction` (SI), `Paralinguistics` (PARA), `Audio Captioning` (AC), and `Audio Scene Question Answering` (ASQA).
|
34 |
|
35 |
[More information about the 8 tasks and evaluation results]
|
36 |
|
|
|
136 |
|
137 |
### Compute and Infrastructure
|
138 |
|
139 |
+
MERaLiON-AudioLLM is trained on the **ASPIRE 2A+** Supercomputer Cluster, provided by the **National Supercomputing Centre (NSCC)**. ASPIRE 2A+ cluster provides multiple H100 nodes, with each compute node equipped with 8 Nvidia H100 GPUs, 2 TB of RAM, and 30 TB of locally attached NVMe storage. These nodes are interconnected via a rail-optimised, full fat-tree topology, utilising 400 Gb/s NDR InfiniBand cables. Additionally, the cluster incorporates a 2.5 PB SSD-based Lustre file system, linked to the H100 nodes through high-speed InfiniBand connections.
|
140 |
|
141 |
With a global batch size of 640, we train the current release of MERaLiON-AudioLLM for around 200k steps, which took 2 days to complete using 16 nodes, 128 H100 GPUs.
|
142 |
|