DOCTOR: A Multi-Disease Detection Continual Learning Framework Based on Wearable Medical Sensors

05/09/2023
by   Chia-Hao Li, et al.
0

Modern advances in machine learning (ML) and wearable medical sensors (WMSs) in edge devices have enabled ML-driven disease detection for smart healthcare. Conventional ML-driven disease detection methods rely on customizing individual models for each disease and its corresponding WMS data. However, such methods lack adaptability to distribution shifts and new task classification classes. Also, they need to be rearchitected and retrained from scratch for each new disease. Moreover, installing multiple ML models in an edge device consumes excessive memory, drains the battery faster, and complicates the detection process. To address these challenges, we propose DOCTOR, a multi-disease detection continual learning (CL) framework based on WMSs. It employs a multi-headed deep neural network (DNN) and an exemplar-replay-style CL algorithm. The CL algorithm enables the framework to continually learn new missions where different data distributions, classification classes, and disease detection tasks are introduced sequentially. It counteracts catastrophic forgetting with a data preservation method and a synthetic data generation module. The data preservation method efficiently preserves the most informative subset of training data from previous missions based on the average training loss of each data instance. The synthetic data generation module models the probability distribution of the real training data and then generates as much synthetic data as needed for replays while maintaining data privacy. The multi-headed DNN enables DOCTOR to detect multiple diseases simultaneously based on user WMS data. We demonstrate DOCTOR's efficacy in maintaining high multi-disease classification accuracy with a single DNN model in various CL experiments. DOCTOR achieves very competitive performance across all CL scenarios relative to the ideal joint-training framework while maintaining a small model size.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
08/03/2023

Efficient Model Adaptation for Continual Learning at the Edge

Most machine learning (ML) systems assume stationary and matching data d...
research
04/12/2022

LifeLonger: A Benchmark for Continual Disease Classification

Deep learning models have shown a great effectiveness in recognition of ...
research
07/13/2023

A Scenario-Based Functional Testing Approach to Improving DNN Performance

This paper proposes a scenario-based functional testing approach for enh...
research
08/20/2022

A Multi-Head Model for Continual Learning via Out-of-Distribution Replay

This paper studies class incremental learning (CIL) of continual learnin...
research
07/15/2022

Continual Learning For On-Device Environmental Sound Classification

Continuously learning new classes without catastrophic forgetting is a c...
research
08/31/2023

Continual Learning From a Stream of APIs

Continual learning (CL) aims to learn new tasks without forgetting previ...
research
02/19/2022

Improving the Level of Autism Discrimination through GraphRNN Link Prediction

Dataset is the key of deep learning in Autism disease research. However,...

Please sign up or login with your details

Forgot password? Click here to reset