現在AI人工智能這么發達,小編們每天都在擔心AI會把寫文章這事給取代了,沒想到現在真的可以了(誤),B站UP主爆肝100天就開發出了一個能寫高考作文的AI程序。
日前B站UP主圖靈的貓演示了一個新的AI程序EssayKiller(作文殺手),它是基于OCR、NLP領域的最新模型所構建的生成式文本創作AI框架。
圖靈的貓表示,目前第一版finetune模型針對高考作文(主要是議論文),可以有效生成符合人類認知的文章,多數文章經過測試可以達到正常高中生及格作文水平。
Essaykiller由4個神經網絡組成,具體作用如下:
第一部分的識別網絡,通過外接攝像頭的實時OCR以及卷積神經網絡進行漢字的識別與提取。
第二部分語言網絡分為兩個子網絡,一個網絡負責讀題能力,另一個負責寫作能力。后者為整改AI的核心。
第三部分為判分網絡,取出Essaykiller生成的多篇文章中,打分最高的一篇文章進行輸出。
為了提高作文能力,Essaykiller共學習了360萬張圖像數據、2億條中文預訓練語料、上千篇微調文章。
期間由于訓練任務重,他自己的RTX 2060 Super顯卡都撐不住了,主要是Essaykiller的神經網絡達到了17億之多,已經超越了目前市面上可見的任何一個單GPU的運算內存限制。
最終找到的解決方法是上NVIDIA的RTX 8000顯卡,它擁有48GB顯存,是目前世界上最大單卡顯存的顯卡。
感興趣的了解下Github開源介紹。