近日，Two Six Labs和斯坦福研究團(tuán)隊(duì)利用神經(jīng)網(wǎng)絡(luò)實(shí)現(xiàn)了對(duì)小鼠的“讀心術(shù)”，他們利用網(wǎng)絡(luò)模型讀取小鼠腦內(nèi)的電信號(hào)，預(yù)測(cè)小鼠的行為和在迷宮中的位置，平均預(yù)測(cè)誤差僅為4cm。

“讀心術(shù)”真的能夠?qū)崿F(xiàn)嗎？

近日，由DARPA和斯坦福的研究團(tuán)隊(duì)正在研究如何“讀小鼠的心”。當(dāng)然，其實(shí)沒(méi)有“讀心術(shù)”那么玄乎，確切地說(shuō)，是通過(guò)神經(jīng)網(wǎng)絡(luò)讀取小鼠大腦中的電信號(hào)活動(dòng)，來(lái)預(yù)測(cè)小鼠的活動(dòng)和位置。

讀取小鼠的“想法”，預(yù)測(cè)小鼠的位置

大腦由相互連接的神經(jīng)元組成：神經(jīng)元可以響應(yīng)輸入處于激活狀態(tài)，反過(guò)來(lái)激活其他神經(jīng)元。這些系統(tǒng)的“簡(jiǎn)化版”就是第一個(gè)人工神經(jīng)網(wǎng)絡(luò)的靈感來(lái)源。斯坦福Schnitzer實(shí)驗(yàn)室的同事們制作了一個(gè)數(shù)據(jù)集，用于監(jiān)控實(shí)驗(yàn)室的小鼠在“競(jìng)技場(chǎng)”中移動(dòng)時(shí)的神經(jīng)活動(dòng)。

所謂“競(jìng)技場(chǎng)”其實(shí)是一個(gè)帶有地標(biāo)貼紙的小盒子。研究人員通過(guò)將一個(gè)微型顯微鏡連接到小鼠的頭部，并記錄熒光染料的軌跡，這種染料會(huì)在單個(gè)神經(jīng)元在放電時(shí)發(fā)出綠光，從而實(shí)現(xiàn)記錄神經(jīng)活動(dòng)的目的。這項(xiàng)技術(shù)可以同時(shí)跟蹤數(shù)百個(gè)、甚至數(shù)千個(gè)神經(jīng)元的活動(dòng)。

我們主要關(guān)注小鼠大腦中海馬體CA1區(qū)域的神經(jīng)元，這是大腦中涉及學(xué)習(xí)、記憶和導(dǎo)航的部分。該區(qū)域中的一些神經(jīng)元被稱(chēng)為“放置細(xì)胞”，因?yàn)樗鼈冺憫?yīng)于鼠標(biāo)的位置而發(fā)射。例如，當(dāng)鼠標(biāo)位于機(jī)箱的左上角時(shí)，給定的單元格可能只會(huì)觸發(fā)。鼠標(biāo)的大腦通過(guò)解釋這些細(xì)胞活動(dòng)或不活動(dòng)的組合信號(hào)來(lái)編碼位置概念。

“可以使用人工神經(jīng)網(wǎng)絡(luò)將這些生物神經(jīng)元的信號(hào)標(biāo)記在小鼠所處位置的地圖上嗎？”也就是說(shuō)，如果我們對(duì)生物神經(jīng)網(wǎng)絡(luò)進(jìn)行逆向工程，是否可以通過(guò)讀取小鼠的意念得知它的位置？

準(zhǔn)確預(yù)測(cè)生物神經(jīng)元活動(dòng)的位置

為此我們訓(xùn)練了一個(gè)神經(jīng)網(wǎng)絡(luò)，根據(jù)最近的神經(jīng)元放電模式預(yù)測(cè)小鼠的位置。我們使用實(shí)驗(yàn)觀察結(jié)果的前80%作為訓(xùn)練數(shù)據(jù)，僅給出神經(jīng)元的活動(dòng)，來(lái)預(yù)測(cè)后20％觀察結(jié)果的小鼠位置。我們嘗試了許多模型體系結(jié)構(gòu)，發(fā)現(xiàn)具有回歸輸出層的簡(jiǎn)單密集神經(jīng)網(wǎng)絡(luò)表現(xiàn)最好，平均預(yù)測(cè)誤差僅為4 cm。小鼠身長(zhǎng)約8厘米，而競(jìng)技場(chǎng)大小為45cm×60cm的矩形。此循環(huán)動(dòng)畫(huà)中顯示了我們的預(yù)測(cè)（藍(lán)點(diǎn)）和小鼠的標(biāo)記位置（紅點(diǎn)）。

模型預(yù)測(cè)給出的位置（藍(lán)點(diǎn)）和小鼠的標(biāo)記位置（紅點(diǎn)）

不過(guò)，盡管回歸輸出表現(xiàn)良好，但沒(méi)有表現(xiàn)出對(duì)其他預(yù)測(cè)的確定性的任何信息。為此我們?cè)O(shè)計(jì)了另一個(gè)深度神經(jīng)網(wǎng)絡(luò)模型，這次的模型包括卷積層。我們將“競(jìng)技場(chǎng)”劃分為1厘米見(jiàn)方的網(wǎng)格，并訓(xùn)練分類(lèi)任務(wù)，預(yù)測(cè)小鼠將走過(guò)“競(jìng)技場(chǎng)”中的哪些網(wǎng)格方塊。模型為預(yù)測(cè)了小鼠會(huì)經(jīng)過(guò)每個(gè)方塊的概率，輸出了一張預(yù)測(cè)強(qiáng)度的熱圖。

但是，由于小鼠的實(shí)際位置的標(biāo)簽是單個(gè)網(wǎng)格方塊（以小鼠的中心點(diǎn)為準(zhǔn)），我們需要開(kāi)發(fā)一種新的損失函數(shù)來(lái)訓(xùn)練模型，告訴模型“幾乎正確”的預(yù)測(cè)比“根本不貼邊”的預(yù)測(cè)更好。之后，模型的表現(xiàn)與點(diǎn)預(yù)測(cè)模型基本相當(dāng)，平均預(yù)測(cè)誤差為5厘米。但是，預(yù)測(cè)信息中包含了有關(guān)替代預(yù)測(cè)和模型確定性的更多信息。視頻中的藍(lán)云表示競(jìng)技場(chǎng)中小鼠所在位置預(yù)測(cè)概率最高的區(qū)域。

藍(lán)色云代表競(jìng)技場(chǎng)中小鼠所在的預(yù)測(cè)概率最高的區(qū)域。紅點(diǎn)是小鼠的標(biāo)記位置

預(yù)測(cè)未來(lái)行為：通過(guò)小鼠行為，破解人類(lèi)行為的奧秘

我們將這種不確定性概念編入預(yù)測(cè)模型中，然后研究：“我們可以通過(guò)讀取小鼠的思想，來(lái)預(yù)測(cè)其未來(lái)的位置嗎？新模型不會(huì)通過(guò)查看最近的神經(jīng)元放電模式，詢問(wèn)小鼠目前的位置，而是預(yù)測(cè)1秒，2秒或3秒后的位置。預(yù)測(cè)結(jié)果與基線相比仍然有很好的表現(xiàn)。

我們分析的數(shù)據(jù)可以表示出小鼠的簡(jiǎn)單行為：在盒子中移動(dòng)。我們可以將數(shù)據(jù)劃分為兩種行為類(lèi)別：“活躍”/移動(dòng)，或固定/“靜止”。我們能夠以75％的平衡精度預(yù)測(cè)小鼠的當(dāng)前行為屬于哪一類(lèi)，并且在2秒后仍然具有66％的準(zhǔn)確度。這表明我們的海馬神經(jīng)元模型不僅能夠編碼有關(guān)現(xiàn)在的位置信息，還對(duì)未來(lái)進(jìn)行某種程度的規(guī)劃。

我們?cè)赟chnitzer實(shí)驗(yàn)室的合作者正在努力制作更復(fù)雜的行為數(shù)據(jù)集，以便我們應(yīng)用這些方法。比如可以在小鼠通過(guò)迷宮時(shí)對(duì)其進(jìn)行映射，預(yù)測(cè)左右轉(zhuǎn)彎，并量化小鼠在學(xué)習(xí)走迷宮時(shí)的不確定性。或識(shí)別對(duì)小鼠展示哪些主題的圖像會(huì)刺激到它。我們使用小鼠作為研究模型，目的是更多地了解我們自己，希望我們的人工神經(jīng)網(wǎng)絡(luò)有助于更好地理解生物的神經(jīng)網(wǎng)絡(luò)。