模型合并就進化，直接拿下SOTA！Transformer作者創(chuàng)業(yè)新成果火了

豐色 2024-03-25 13:41:40 來源：量子位

豐色 發(fā)自 凹非寺

量子位 | 公眾號 QbitAI

把Huggingface上的現(xiàn)成模型拿來“攢一攢”——

直接就能組合出新的強大模型？！

日本大模型公司sakana.ai腦洞大開（正是“Transformer八子”之一所創(chuàng)辦的公司），想出了這么一個進化合并模型的妙招。

該方法不僅能自動生成新的基礎(chǔ)模型，而且性能絕不賴：

他們得到的一個70億參數(shù)的日語數(shù)學(xué)大模型，直接在相關(guān)基準(zhǔn)測試上取得了SOTA，打敗了700億參數(shù)的Llama-2等前輩。

最重要的是，得出這樣的模型不需要任何梯度訓(xùn)練，因此需要的計算資源大大減少。

英偉達科學(xué)家Jim Fan看完大贊：

這是我最近讀過的最有想象力的論文之一。

合并進化，自動生成新基礎(chǔ)模型

從開源大模型排行榜上表現(xiàn)最好的模型，大多不再是LLaMA或Mistral這種“原始”模型，而是一些微調(diào)或合并模型之后，我們就能看出：

一種新的趨勢出現(xiàn)了。

Sakana.ai介紹，開源基礎(chǔ)模型很容易在數(shù)百個不同的方向上進行擴展和微調(diào)，然后產(chǎn)生在新的領(lǐng)域表現(xiàn)出色的新模型。

而在這之中，模型合并顯現(xiàn)出了巨大前景。

但，它可能是一種“黑魔法”，嚴(yán)重依賴直覺和專業(yè)知識。

因此，我們需要更為系統(tǒng)性的方法。

受自然界的自然選擇啟發(fā)，Sakana.ai盯上了進化算法，引入“進化模型合并”（Evolutionary Model Merge）的概念，提出一種可以發(fā)現(xiàn)最佳模型組合的通用方法。

該方法結(jié)合了兩種不同的思路：

（1）合并數(shù)據(jù)流空間（層）中的模型，以及（2）合并參數(shù)空間（權(quán)重）中的模型。

具體而言，第一種數(shù)據(jù)流空間方法是通過進化來發(fā)現(xiàn)不同模型層的最佳組合，以此形成新模型。

在社區(qū)以往的做法中，都是靠直覺來確定如何以及模型哪些層可以與另一個模型的層結(jié)合。

但其實，Sakana.ai介紹，這個問題有一個組合數(shù)量巨大的搜索空間，最適合由優(yōu)化算法如進化算法來搜索。

其操作示例如下：

至于第二個參數(shù)空間方法則混合多個模型權(quán)重來形成新模型。

這種方法其實很無數(shù)種實現(xiàn)，再加上混合的每一層原則上可以使用不同的混合比例，就更多了。

而這，利用進化方法就可以有效地找出更為新穎的混合策略。

以下是將兩個不同模型的權(quán)重進行混合得到新模型的操作示例：

將以上這兩種方法合并，就是這樣的：

作者介紹，他們希望在相距較遠的領(lǐng)域，例如數(shù)學(xué)和非英語語言、視覺和非英語語言，來組成之前大家不曾探索過的新興組合。

結(jié)果，還真有點讓人驚喜。

新模型輕松拿下SOTA

用以上進化合并方法，團隊得到了3個基礎(chǔ)模型：

• 大語言模型EvoLLM-JP

由日語大模型Shisa-Gamma和數(shù)學(xué)大模型WizardMath/Abel合并而成，擅長解決日語數(shù)學(xué)問題，進化了100-150代。

• 視覺語言模型EvoVLM-JP

日語大模型Shisa Gamma 7B v1+LLaVa-1.6-Mistral-7B，是具有日語能力的VLM。

• 圖像生成模型EvoSDXL-JP

支持日語的SDXL擴散模型。

前兩個已在Hugging Face和GitHub上發(fā)布，最后一個也即將推出。

具體來看。

1、EvoLLM-JP

它在GSM8K數(shù)據(jù)集的多語言版本——MGSM的日語評估集上取得成績?nèi)缦拢?/p>

可以看到，EvoLLM-JP用日語解決數(shù)學(xué)問題的表現(xiàn)超過了它們的原始模型，也超過了Llama-2、GPT-3.5等高性能模型。

其中模型4是僅在參數(shù)空間進行了優(yōu)化，模型6是使用模型4在數(shù)據(jù)流空間中進一步優(yōu)化的結(jié)果。

在既評估數(shù)據(jù)能力也評估一般日語能力的日語lm-evaluation-harness基準(zhǔn)上，EvoLLM-JP則在9個任務(wù)上的平均得分最高達到了70.5——只用70億參數(shù)，它就打敗了700億的Llama-2等模型。

團隊表示，EvoLLM-JP已經(jīng)足夠優(yōu)秀，可以作為通用日語大模型，并解決一些有趣的例子：

比如需要特定日本文化知識的數(shù)學(xué)問題，或者用關(guān)西方言講日本笑話。

2、EvoVLM-JP

在以下兩個圖像問答的基準(zhǔn)數(shù)據(jù)集上，分數(shù)越高，代表模型用日語回答的描述越準(zhǔn)確。

結(jié)果，它不僅比其所基于的英語VLM LLaVa-1.6-Mistral-7B更出色，也比現(xiàn)有的日語VLM更厲害。

如下圖所示，在回答圖中的信號燈為什么顏色之時，只有EvoVLM-JP答對：藍色。（日本的習(xí)俗就是把紅綠燈稱為紅藍燈）

3、EvoSDXL-JP

這個支持日語的SDXL模型只需4個擴散模型即可執(zhí)行推理，生成速度相當(dāng)快。

具體跑分還沒出來，但團隊透露也是“相當(dāng)有希望的”。

可以欣賞一些示例：

提示詞包括：味噌ラーメン、最高品質(zhì)の浮世絵、葛飾北斎、江戸時代。

對于以上3個新模型，團隊指出：

原則上，我們可以采用基于梯度的反向傳播來進一步提高以上這些模型的性能。

但我們不用，因為現(xiàn)在的目的就是表明，即使沒有反向傳播，我們?nèi)匀豢梢缘玫阶銐蛳冗M的基礎(chǔ)模型，挑戰(zhàn)當(dāng)前的“昂貴范式”。

對此，網(wǎng)友們紛紛點贊。

Jim Fan也補充：

在基礎(chǔ)模型領(lǐng)域，目前社區(qū)幾乎完全專注于讓模型去學(xué)習(xí)，而不太重視搜索，但后者在訓(xùn)練（也就是本文提出的進化算法）和推理階段其實都有巨大的潛力。

所以，如網(wǎng)友所說：

我們現(xiàn)在已經(jīng)處于模型的寒武紀(jì)大爆發(fā)時代了嗎？

論文地址：
https://arxiv.org/abs/2403.13187
參考鏈接：
[1]https://sakana.ai/evolutionary-model-merge/
[2]https://twitter.com/DrJimFan/status/1771927650883522899?s=20
[3]https://twitter.com/SakanaAILabs/status/1770613032198279663