Democratic learning: hardware/software co-design for lightweight blockchain-secured on-device machine learning

• Published in: Journal of systems architecture 2021-09, Vol.118, p.102205, Article 102205
• Main Authors: Zhang, Rui, Song, Mingcong, Li, Tao, Yu, Zhibin, Dai, Yuting, Liu, Xiaoguang, Wang, Gang
• Format: Article
• Language: English
• Subjects: Accelerator; Application-specific system; Architecture/software co-design; Emerging computing platform; Privacy support; Security support
• ISSN: 1383-7621; 1873-6165
• DOI: 10.1016/j.sysarc.2021.102205

Recently, the trending 5G technology encourages extensive applications of on-device machine learning, which collects user data for model training. This requires cost-effective techniques to preserve the privacy and the security of model training within the resource-constrained environment. Traditional learning methods rely on the trust among the system for privacy and security. However, with the increase of the learning scale, maintaining every edge device’s trustworthiness could be expensive. To cost-effectively establish trust in a trustless environment, this paper proposes democratic learning (DemL), which makes the first step to explore hardware/software co-design for blockchain-secured decentralized on-device learning. By utilizing blockchain’s decentralization and tamper-proofing, our design secures AI learning in a trustless environment. To tackle the extra overhead introduced by blockchain, we propose PoMC (an algorithm and architecture co-design) as a novel blockchain consensus mechanism, which first exploits cross-domain reuse (AI learning and blockchain consensus) in AI learning architecture. Evaluation results show our DemL can protect AI learning from privacy leakage and model pollution, and demonstrated that privacy and security come with trivial hardware overhead and power consumption (2%). We believe that our work will open the door of synergizing blockchain and on-device learning for security and privacy.