0
就像程序員討論“什么才是世界上最好的語言？”這一終極命題一樣。在醫(yī)療AI領域，圍繞芯片和AI開源框架的討論也一直熱度不減。
《后浪》視頻一夜刷屏。
其實在AI時代，芯片和開源框架也有諸多不斷趕超“前浪”的攪局者和顛覆者。
為了搶占市場、營造生態(tài)，科技巨頭也都推出了自己的開源深度學習框架，如Google的TensorFlow、Amazon的MXNet、Facebook的Pytorch和Microsoft的CNTK等。
而在國內，也有百度飛獎、曠視MegEngine等框架進入越來越多開發(fā)者的視野。
但這個行業(yè)無時不刻都在洗牌中。
有開發(fā)者向雷鋒網表示，大名鼎鼎的Caffe因其靈活性等相對較差，導致用戶逐年流失。雖然Caffe2在一定程度上進行了改善，但是已經錯過了重要的框架發(fā)展窗口期。
新勢力也逐漸崛起。
今年4月底，英偉達也與倫敦國王學院共同宣布了Project MONAI的開源alpha版本。MONAI是AI Medical Open Network的縮寫。這個框架針對醫(yī)療保健研究人員的需求進行了優(yōu)化，并且可以與PyTorch和Ignite等深度學習框架一起運行。
這個框架推出之時，國內就有學者評價到，“為醫(yī)學影像研究者量身打造，是真正的良心之作?！?/p>
當然，除了開源框架，AI芯片也一直都是行業(yè)巨頭酣戰(zhàn)的領域，英特爾、英偉達以及眾多AI專用芯片公司的醫(yī)療AI相關產品層出不窮。
作為開源框架和芯片的“首席體驗官”，算法工程師們如何看待芯片、開源框架等領域的諸多選擇，我們采訪了四家主流醫(yī)療AI公司的技術人員。
柏視醫(yī)療董事長陸遙：快速驗證模型效果使用Pytorch，工業(yè)化部署使用TensorFlow。
TensorFlow： TensorFlow提供全面的服務，無論是Python，C++，JAVA，Go，甚至是JavaScript，Julia，C#；TensorFlow良好的文檔和社區(qū)支持，大大降低了學習成本；提供直觀的可視化計算圖；模型能夠快速的部署在各種硬件機器上，從高性能的計算機到移動設備，再到更小的更輕量的智能終端。但構建TensorFlow深度學習框架需要更復雜的代碼，還要重復地多次構建靜態(tài)圖。不過到２.0以后，tensorflow將keras融合進來，很多操作都簡化了。
Pytorch：Pytorch的風格和python很像，使用的是動態(tài)圖，搭建模型比較快，方便調試，能夠很快驗證方法的有效性，新手上手快，而且?guī)熳銐蚝唵?，跟NumPy、SciPy等可以無縫連接；設計簡單，動態(tài)地設計網絡，而無需定義靜態(tài)網絡圖，非常靈活。但pytorch在生產環(huán)境部署還不成熟。
Caffe：Caffe 是一個優(yōu)先考慮表達、速度和模塊化來設計的框架。支持 C、C++、Python等接口以及命令行接口。但是，Caffe 不支持精細粒度網絡層，給定體系結構，對循環(huán)網絡和語言建模的總體支持相當差，必須用低級語言建立復雜的層類型，使用門檻很高。雖然開源比Pytorch和Tensorflow要早一點，早期的很多機器視覺的算法是基于Caffe框架，但是Caffe社區(qū)比較前兩個框架要小很多，更新迭代的也比較慢。
連心醫(yī)療算法經理Fisher：我們目前用的主要是Tensorflow以及依據Tensorflow的Keras框架，選擇Tensorflow的主要原因是市場占有率大、社區(qū)支持好，而且最主要的是工業(yè)部署非常成熟。
Pytorch現在研究人員用的比較多，最主要的優(yōu)點是其動態(tài)圖的實現方案，對于測試等非常友好。但是，在工業(yè)部署上，和Tensorflow還是有一定差距。目前，這兩個框架選擇的人比較多。
Caffe用的人已經在逐漸減少，主要還是其靈活性等相對較差，雖然Caffe2在一定程度上進行了改善，但是已經錯過了重要的框架發(fā)展窗口期。
聯影智能研發(fā)科學家：相比其他兩個，Pytorch在算法實現靈活度更強，比較貼近機器學習研發(fā)工程師。同時學術界對Pytorch的支持也較為強大，整個圈子比較活躍。因為我們部署這塊都是采用內部開發(fā)的部署框架，所以我們更為看中訓練研發(fā)階段框架的靈活性和高效性。
匯醫(yī)慧影算法總監(jiān)劉鵬飛：Pytorch上手難度低，調試方便，代碼寫起來比較pythonic，但目前對部署的支持相對較弱，TensorFlow對部署支持比較好，但相對上手難度高，調試困難。Caffe是第一代深度學習框架，雖然仍然用于成熟模型部署，但基于Caffe的模型研發(fā)成本比較高，比如，很多情況下用戶需要自己實現反向傳播代碼，不適用于快速模型開發(fā)和迭代。
柏視醫(yī)療董事長陸遙：國內的開源框架，比如百度的飛槳和最近推出的清華大學Jittor、曠視的MegEngine和華為的MindSpore，這都是很好的嘗試，但是國內開源框架都面臨和Caffe一樣的問題，使用者較少，資源比Pytorch、Tensorflow少很多，和Pytorch和Tensorflow框架的兼容性不好，在這些框架上做開發(fā)的難度相對比較大。因此，國內這些框架的社區(qū)還是比較小。
百度在飛槳的推廣上做了很多的工作，可能也發(fā)現在通用框架上，和這些國外成熟框架競爭所面臨的困境。近期推出的影像AI專用框架，是一個不錯的想法，因為Pytorch和Tensorflow這類框架是針對通用性設計的，在細分領域并不算非常好。如果影像AI這一塊，飛槳能做的足夠出色，吸引更多的研發(fā)者進來，做好生態(tài)，也能形成自己的社區(qū)，希望國內開源框架的生態(tài)越做越好。
連心醫(yī)療算法經理Fisher：我們一直和飛槳有合作，在春節(jié)期間還一起推出了新冠肺炎的檢測模型。
最近除了百度，華為以及曠視等也都推出了自己的框架，百度應該是國內開發(fā)最早、發(fā)展最成熟的框架。
國內框架的優(yōu)點是中文的支持友好，和企業(yè)直接合作較多以及硬件的直接融合開發(fā)等，但是致命的缺點在于其學術界的缺位。
放眼最新的論文，其開源的代碼基本都是Tensorflow以及PyTorch的實現。當然如果是比較火熱的論文，框架一般也都會及時把相應的代碼跟上，但是如果不是大熱的論文，都需要自己再復現一下論文。
在目前工程師基本都要掌握Tensorflow和Pytorch兩種框架的情景下，讓他們再去掌握第三種框架是缺乏動力的。
聯影智能研發(fā)科學家：基本沒有太多接觸，主要還是基于飛槳的應用以及開源代碼，資料這方面比較少，暫時沒有Pytorch和Tensorflow生態(tài)圈龐大。所以無論上手，學習，成本都會相應增高。
柏視醫(yī)療董事長陸遙：MONAI使用PyTorch深度學習框架，旨在提供一種開源、標準化程度高、用戶友好、可復現性好、易于集成、高質量的、針對醫(yī)療領域特定優(yōu)化的深度學習框架。
從長期來看，這是趨勢，但需要吸引更加多的開發(fā)者加入，壯大社區(qū)的力量。
它和飛槳影像AI是一個思路，走的是細分領域，依托于英偉達自身的硬件基礎，在醫(yī)療AI影像上，可以做出更出色的性能。在醫(yī)療AI領域，和Pytorch兼容，減少Pytorch用戶遷移到MONAI的難度，提高用戶對MONAI的使用率，可以實現和Pytorch共同發(fā)展。
MONAI在計算效率、并行化訓練和部署，以及醫(yī)療特定任務的支持上應該會有更好的效果，而Pytorch是一個更基礎通用的框架。二者各有所長，未來應該是相互促進的關系。
連心醫(yī)療算法經理Fisher：簡單的總結就是，不會。
首先，我們要搞清楚MONAI是什么，它在Github上的解釋是：MONAI is a PyTorch-based, open-source framework for deep learning in healthcare imaging。
也就是說，首先它和Pytorch并不是排他的關系，MONAI就是基于Pytorch構建的，包括官方的一些例子，都是會引入Torch，也就沒有了什么取代。
在我看來，MONAI更像是一個基于Pytorch的醫(yī)療圖像數據處理庫，針對醫(yī)療圖像增添了許多圖像處理的方法，彌補了Pytorch對于醫(yī)療圖像的支持不足。
柏視醫(yī)療董事長陸遙：服務器的選購主要是考慮成本以及售后服務，在滿足基本配置要求的前提下，選用價格成本較低，售后服務較為完善的服務器供應商。NVIDIA提供一站式解決方案，且計算性能優(yōu)秀，是很好的選擇。
連心醫(yī)療算法經理Fisher：目前，我們服務器供應商有多家，其中DELL的機器較多，主要的原因還是DELL在端機上成熟的解決方案、穩(wěn)定的品質。
聯影智能研發(fā)科學家：目前購買過AMAX、NVIDIA DGX系列，前者用于部署、后者用于訓練，各有好處。NVIDIA DGX在整體解決方案上更為成熟，技術支持也會好些。
匯醫(yī)慧影算法總監(jiān)劉鵬飛：服務器基本就是NVIDIA GPU + Intel CPU。用戶基數大，社區(qū)比較成熟，遇到問題可以很容易找到解決方案。
柏視醫(yī)療董事長陸遙：英偉達。
連心醫(yī)療算法經理Fisher：據我所知，在端側基本只能用英偉達的芯片，因為各家的框架都是構建于CUDA之上，而只有英偉達的卡才可以使用CUDA。
聯影智能研發(fā)科學家：現在市面主流還是NVIDIA，無論是CUDA還是深度學習，NVIDIA的顯卡對各個框架的支持都很好，而且可以兼容三維渲染、并行編程，這方面較其他芯片來說優(yōu)勢目前還是很大的。
匯醫(yī)慧影算法總監(jiān)劉鵬飛：這兩家的AI芯片我們都在用，目前實時性要求比較高應用主要還是基于NVIDIA芯片，計算量相對少的應用一般基于英特爾的芯片。
柏視醫(yī)療董事長陸遙：英偉達系列顯卡最早能夠支持諸如Tensorflow 、Pytorch一類的深度學習框架的部署，技術起步早，國內外相關的研究者比較多，技術生態(tài)比較成熟。
英偉達還有專門針對深度學習優(yōu)化的工具tensorRT，能夠更好地結合深度學習框架與顯卡硬件資源之間的優(yōu)勢，充分發(fā)揮顯卡的計算性能。CUDA很好地封裝了一些對顯卡硬件資源調度和使用的API接口，能夠讓研發(fā)人員方便高效地對顯卡的計算資源進行利用，上手難度較低。
連心醫(yī)療算法經理Fisher：正如上邊所說，雖然有一些解決方案也可以不用CUDA而使用OpenCL，但是基本工業(yè)界只有CUDA這一個選項，這主要是其統一的開發(fā)套件、非常豐富的庫以及NVCC所具備的PTX代碼生成以及離線編譯等更成熟的編譯器特性。另外，眾多的開源框架也都是建立在CUDA之上。
聯影智能研發(fā)科學家：CUDA有很長的一段發(fā)展歷史，沉淀比較多，資料也比較多，優(yōu)勢很大。特別是現在除了CUDA C，還有Thrust這種C++庫，選擇余地大。另外，CUDNN也可以和CUDA一起兼容，所以除了深度學習，還可以做其他GPGPU并行編程，整體就非常方便。
匯醫(yī)慧影算法總監(jiān)劉鵬飛：對于性能要求比較高的場景，英偉達硬件軟件包是比較有優(yōu)勢的。CUDA、TensorRT等工具都是經過高度優(yōu)化，執(zhí)行速度上有比較明顯的優(yōu)勢。
柏視醫(yī)療董事長陸遙：醫(yī)療AI專用芯片，具有非常巨大的市場空間，人工智能技術的快速發(fā)展，讓AI算法在醫(yī)療領域已經得到了行業(yè)認可，但醫(yī)療行業(yè)有其行業(yè)特點：數據量大、高維度、多模態(tài)，顯存更大的芯片能夠加速AI算法落地復雜的醫(yī)療環(huán)境。
對醫(yī)療AI專用芯片的選擇，內存是考慮因素之一，同時計算力也是一個考量的因素，在這點上英偉達更有優(yōu)勢。所以選擇上，看個人更看重哪個因素。
英特爾的醫(yī)療AI專用芯片技術起步比較晚，在技術生態(tài)環(huán)境上暫時比不上英偉達，研發(fā)人員對英特爾系列的AI芯片的應用有一定的上手難度。對英特爾芯片的應用，很多技術人員還處于摸索階段。從研發(fā)效率和穩(wěn)定性來考慮。目前，選擇英偉達的芯片是大多數的選擇。
更大顯存對醫(yī)療AI領域確實有著一定的優(yōu)勢。隨著技術的發(fā)展，使用英特爾的醫(yī)療AI專用芯片做部署環(huán)境也是可以考慮的技術路線之一。
連心醫(yī)療算法經理Fisher：從我個人角度來看，硬件是服務于項目的，而項目中硬件只是制約的一部分因素，更重要的是開發(fā)效率。誠然，顯存是會讓算法有更多的可能性，但是開發(fā)效率會是工程上更關注的部分。
如果沒有良好的社區(qū)支持以及市場支持，再好的硬件也只能是和一些特定廠商的定制硬件選項而不是廣泛的工程選擇。就像顯卡上NVIDIA和AMD的競爭，由于CUDA的存在以及已經發(fā)展起來的良好社區(qū)，如果想要進行科學計算，那么可以說，英偉達是唯一的選項。
匯醫(yī)慧影算法總監(jiān)劉鵬飛：據我所知，目前英特爾的AI芯片主要應用在推理階段。在訓練階段，大家基本還是使用NVIDIA GPU。另外，對3D醫(yī)療圖像來說，Intel OpenVino目前還有一些常用算子沒有完全支持。
雷峰網原創(chuàng)文章，未經授權禁止轉載。詳情見轉載須知。
