AI Hardware Accelerator Design based on Prognosis Prediction Deep Learning Network for Cancer Risk Stratification
암 예후 예측 딥러닝 네트워크 기반 AI 하드웨어 가속기 설계
초록/요약
With the recent progress of artificial intelligence (AI) technology, deep learning-based approaches in the medical field have increased lately. This paper proposes a deep learning network using Ajou University Hospital’s 10-year breast cancer patient dataset to predict the recurrence year of cancer. The proposed network analyzes the whole prognostic factors of the patient. In addition, the influence of each prognostic factor was analyzed by excluding the factors in the training respectively. The network showed high performance by achieving 0.91 area under the receiver operating characteristic (ROC) curve (AUC). For AI hardware accelerator implementation, the proposed fixed 16-bit integer quantization method was performed to compress the parameter of the proposed network. The proposed quantization method enabled 37.41% parameter compression of the proposed network. The accelerator showed higher throughput and lower power consumption than the graphic processing unit (GPU). The proposed hardware accelerator architecture is implemented on the Xilinx Kintex UltraScale+ field programmable gate array (FPGA).
more목차
I.Introduction 1
II. Proposed Method 5
A. Proposed Deep Learning Network Architecture 5
B. Data Collection 6
C. Loss Function 9
D. Integer Quantization Method 9
E. AI Hardware Accelerator Architecture 11
F. Hyperparameters and Environment 13
III. Experimental Results 14
A. Statistical Analysis and Evaluation Metrics Method 14
B. Data Characteristics 15
C. Network Evaluation 17
D. Quantization Results and Comparison 26
E. FPGA Implementation Results 27
IV. Conclusion 30
