用优化数据集训练神经网络,让糖尿病胰岛素控制更高效精准
Neural Networks for on-chip Model Predictive Control: a Method to Build Optimized Training Datasets and its application to Type-1 Diabetes
- 构建最优采样数据集,避免重复状态,提升训练效率
- 在糖尿病控制中使神经网络精度提高四倍,接近原算法性能
- 成果已获监管批准,可用于人体临床测试,适合嵌入式医疗设备
将神经网络训练为模型预测控制(MPC)算法是实现在资源受限嵌入式设备中的有效方法。通过收集大量输入-输出数据(输入为系统状态,输出为MPC生成的控制动作),神经网络可低成本复现MPC行为。然而,训练数据的构成显著影响最终精度,而系统化优化方法仍不充分。本文提出理想训练集——最优采样数据集(OSD),并设计高效生成算法。OSD满足三个条件:(i)保留至特定数值分辨率的原有MPC信息,(ii)避免重复或近似重复状态,(iii)达到饱和或完整状态。我们在弗吉尼亚大学的自动胰岛素输送MPC算法上验证了该方法,使神经网络精度提升四倍。两个基于OSD训练的神经网络已获监管机构批准进入临床测试,成为首个用于直接人体胰岛素给药的神经网络控制算法。该方法为资源受限平台部署复杂算法开辟新路径。
原文摘要 · Abstract (English)
Training Neural Networks (NNs) to behave as Model Predictive Control (MPC) algorithms is an effective way to implement them in constrained embedded devices. By collecting large amounts of input-output data, where inputs represent system states and outputs are MPC-generated control actions, NNs can be trained to replicate MPC behavior at a fraction of the computational cost. However, although the composition of the training data critically influences the final NN accuracy, methods for systematically optimizing it remain underexplored. In this paper, we introduce the concept of Optimally-Sampled Datasets (OSDs) as ideal training sets and present an efficient algorithm for generating them. An OSD is a parametrized subset of all the available data that (i) preserves existing MPC information up to a certain numerical resolution, (ii) avoids duplicate or near-duplicate states, and (iii) becomes saturated or complete. We demonstrate the effectiveness of OSDs by training NNs to replicate the University of Virginia's MPC algorithm for automated insulin delivery in Type-1 Diabetes, achieving a four-fold improvement in final accuracy. Notably, two OSD-trained NNs received regulatory clearance for clinical testing as the first NN-based control algorithm for direct human insulin dosing. This methodology opens new pathways for implementing advanced optimizations on resource-constrained embedded platforms, potentially revolutionizing how complex algorithms are deployed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。