Topological persistence guided knowledge distillation for wearable sensor data

• Published in: Engineering applications of artificial intelligence Vol. 130; p. 107719
• Main Authors: Jeon, Eun Som, Choi, Hongjun, Shukla, Ankita, Wang, Yuan, Lee, Hyunglae, Buman, Matthew P., Turaga, Pavan
• Format: Journal Article
• Language: English
• Published: England Elsevier Ltd 01.04.2024
• Subjects: Deep learning; Feature orthogonality; Knowledge distillation; Topological data analysis; Wearable sensor data; Deep learning; Wearable sensor data; Feature orthogonality; Knowledge distillation; Topological data analysis; feature orthogonality; knowledge distillation; wearable sensor data; topological data analysis
• ISSN: 0952-1976; 1873-6769
• DOI: 10.1016/j.engappai.2023.107719

Deep learning methods have achieved a lot of success in various applications involving converting wearable sensor data to actionable health insights. A common application areas is activity recognition, where deep-learning methods still suffer from limitations such as sensitivity to signal quality, sensor characteristic variations, and variability between subjects. To mitigate these issues, robust features obtained by topological data analysis (TDA) have been suggested as a potential solution. However, there are two significant obstacles to using topological features in deep learning: (1) large computational load to extract topological features using TDA, and (2) different signal representations obtained from deep learning and TDA which makes fusion difficult. In this paper, to enable integration of the strengths of topological methods in deep-learning for time-series data, we propose to use two teacher networks — one trained on the raw time-series data, and another trained on persistence images generated by TDA methods. These two teachers are jointly used to distill a single student model, which utilizes only the raw time-series data at test-time. This approach addresses both issues. The use of KD with multiple teachers utilizes complementary information, and results in a compact model with strong supervisory features and an integrated richer representation. To assimilate desirable information from different modalities, we design new constraints, including orthogonality imposed on feature correlation maps for improving feature expressiveness and allowing the student to easily learn from the teacher. Also, we apply an annealing strategy in KD for fast saturation and better accommodation from different features, while the knowledge gap between the teachers and student is reduced. Finally, a robust student model is distilled, which can at test-time uses only the time-series data as an input, while implicitly preserving topological features. The experimental results demonstrate the effectiveness of the proposed method on wearable sensor data. The proposed method shows 71.74% in classification accuracy on GENEActiv with WRN16-1 (1D CNNs) student, which outperforms baselines and takes much less processing time (less than 17 sec) than teachers on 6k testing samples.