探索用于小占用关键词检测的TinyML框架:一个简明概述

2024-08-30 08:20

本文主要是介绍探索用于小占用关键词检测的TinyML框架:一个简明概述,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

摘要

第一部分:引言

第二部分:部署TinyML的常见挑战

第三部分:SF-KWS的不同方法

A. 网络架构(Network Architecture)

B. 学习技术(Learning Techniques)

C. 模型压缩(Model Compression)

D. 注意力感知架构(Attention Awareness Architecture)

E. 特征优化(Feature Optimization)

F. 神经架构搜索(Neural Architecture Search, NAS)

第四部分:实验研究

第五部分:结论

实验方法

实验结果

模型量化

模型部署

实验结论

代码和模型

参考文献


 

 

这篇论文的标题是《Exploring TinyML Frameworks for Small-Footprint Keyword Spotting: A Concise Overview》,作者是Soumen Garai和Suman Samui,来自印度国立技术学院杜尔加布尔分校电子与通信工程系。论文主要探讨了在智能语音激活设备、智能手机和物联网(IoT)应用中,实现小占用(Small Footprint, SF)关键词检测(Keyword Spotting, KWS)的TinyML框架。以下是对论文主要内容的整理:

摘要

  • 论文讨论了在低功耗和有限内存的边缘设备上实现SF-KWS模型的重要性。
  • 通过深度学习的进步,可以在连续的词汇流中识别预定义的单词或关键词。
  • 论文研究了六种不同的技术类别,包括网络架构、学习技术、模型压缩、注意力感知架构、特征优化和神经网络搜索,这些技术适用于开发SF-KWS系统。
  • 实验表明,Tensorflow Lite框架可以压缩模型约69%,同时保持可接受的准确度。

第一部分:引言

  • 论文介绍了语音助手在日常生活中的重要性,以及它们如何通过云基语音识别实现自然通信。
  • 讨论了将连续的音频流发送到云端进行语音识别的能源效率和网络负载问题。
  • 介绍了关键词检测(KWS)的概念,以及它在唤醒设备并触发云端语音识别中的作用。

f7ec83c8c119458dab2e94a73b2f0382.png

第二部分:部署TinyML的常见挑战

 

  • TinyML旨在在低功耗设备上实现优化和紧凑的机器学习模型。
  • 论文讨论了TinyML解决方案的局限性,包括无法在设备上进行模型训练,以及模型大小和推理延迟的优化问题。
  • 介绍了各种TinyML框架,包括Google的TensorFlow Lite、Microsoft的Embedded Learning Library、ARM的CMSIS-NN等。

373bf10b8ab240099983504400f60e2a.png

第三部分:SF-KWS的不同方法

  • 论文根据文献调查,探讨了多种方法和技术来优化KWS系统,包括网络架构、学习技术、模型压缩、注意力感知、特征优化和神经架构搜索。
  • 每种方法都旨在在准确性和资源效率之间取得平衡。

4df8df7baad5446db01e0f41b7a7c74b.png

A. 网络架构(Network Architecture)

  • 深度神经网络(DNN):使用深度学习技术,如深度残差学习技术和扩张卷积,来提高小占用模型的性能。
  • 卷积循环神经网络(CRNN):结合卷积神经网络和循环神经网络,以捕获时间序列数据的特征。
  • 多分支时间卷积模块(MT-Conv):通过不同尺度的时间模式捕获,以最小的内存和计算开销提高效率。
  • Matchbox架构:使用1D时间通道可分离卷积来减少与常规1D卷积相比的模型大小。

B. 学习技术(Learning Techniques)

  • 残差学习:在深层网络中使用快捷连接来增强训练和模型性能。
  • 知识蒸馏(Knowledge Distillation):将大型模型的知识传递给小型模型,以在减少模型大小和计算需求的同时保持性能。
  • 奇异值分解(Singular Value Decomposition):压缩权重矩阵,保留关键信息,减少参数,适用于SF-KWS。
  • 少样本学习(Few-shot Learning, FSL):解决关键词数据有限的挑战,采用原型网络(ProtoNet)概念,在10次样本情况下实现76%的准确度。

C. 模型压缩(Model Compression)

  • 量化(Quantization):降低模型参数的精度,减少内存占用,同时尽量不牺牲性能。
  • 动态量化:使用较低位宽表示神经网络权重和激活,减少内存使用并加速计算。
  • 错误扩散(Error Diffusion):将量化误差分散在语音特征上,允许更高的压缩比率,同时保持关键词检测的准确性。

D. 注意力感知架构(Attention Awareness Architecture)

  • 注意力机制:通过关注输入序列中的关键部分来减少计算负载,并通过自注意力模型、共享权重自注意力和端到端模型提高关键词检测的准确性和效率。

E. 特征优化(Feature Optimization)

  • 特征选择:选择紧凑且相关的语音信号表示,以实现准确有效的关键词识别。
  • 流特征(Streaming Features):使用时间频率表示,最小化维度,并选择能够捕获关键听觉信息的特征。

F. 神经架构搜索(Neural Architecture Search, NAS)

  • NAS:通过创新算法,如NASNet、AmoebaNet、DARTS和MicroNet,自动发现有效的神经网络架构。
  • NASNet:使用强化学习优化各种操作的结构。
  • AmoebaNet:为具有挑战性的场景细化NAS方法。
  • DARTS:使用基于梯度的方法进行架构优化。
  • MicroNet:专注于小型模型的资源效率。

第四部分:实验研究

  • 论文专注于网络架构组,从六个类别中选择不同的架构进行实验。
  • 使用Google Speech Commands数据集对十类不同的类别进行分类。
  • 对比了不同TinyML框架应用到SF-KWS架构上的性能,包括模型大小、推理时间和准确度。

92b6fb1cdd9240eeba34196531db1069.png

第五部分:结论

  • 论文总结了SF-KWS的探索,强调了Tensorflow Lite框架在减少模型大小方面的显著效果,同时保持了较高的准确度。
  • 指出MicroNet-S架构在模型大小和推理时间上适合IoT设备部署。

实验方法

  • 作者选择了多种网络架构,包括深度神经网络(DNN)、卷积神经网络(CNN)、Deathwish-separable Convolution Networks(DS-CNN)和MicroNet模型。
  • 为了更细致地分类和区分这些架构,作者根据它们的尺寸将它们分为小型(S)和大型(L),例如DNN-S、DNN-L、CNN-S、CNN-L等。
  • 实验使用了不同的TinyML框架,如TensorFlow Lite和Edge Impulse,来适配和优化这些架构,使其适合在物联网(IoT)边缘设备上部署。

实验结果

  • 实验结果显示,TensorFlow Lite框架可以将模型大小压缩约69%,同时保持可接受的准确度,并且推理时间显著减少。
  • 特别是TensorFlow Lite的CNN-S和DS-CNN-S模型,它们在目标边缘设备的闪存上易于部署,非常适合IoT边缘设备。
  • Edge Impulse框架显著减少了模型大小,而MicroNet-S模型在低占用设备上在准确度和内存大小之间进行了权衡。

模型量化

  • 作者将Keras软件模型量化为Int8格式,以适应TensorFlow Lite模型,包括MicroNet架构。
  • 量化过程导致了模型大小的显著减少,同时保持了推理的准确性。

模型部署

  • 作者评估了不同框架在Cortex-M板的推理时间、模型大小和最终部署能力。
  • 表IV(Table IV)可视化了不同TinyML框架应用到各种SF-KWS架构的比较评估。

实验结论

  • 实验结果表明,TensorFlow Lite框架在减少模型占用空间方面表现出色,同时MicroNet-S架构在保持高准确度的同时显著减少了推理时间,非常适合部署在IoT设备上。

代码和模型

  • 作者使用了GitHub上的Keras软件模型,并对其进行了训练和评估。
  • EdgeImpulse框架采用了基于Mel-frequency cepstral coefficients (MFCC)和1D卷积的架构,包括float32和int8压缩技术。

参考文献

  • 论文列出了一系列相关的研究文献,涵盖了关键词检测、TinyML框架、模型压缩和神经网络搜索等领域。

论文通过实验和理论分析,为在资源受限的IoT设备上部署SF-KWS模型提供了有价值的见解和框架比较。

 

这篇关于探索用于小占用关键词检测的TinyML框架:一个简明概述的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1120285

相关文章

Python Dash框架在数据可视化仪表板中的应用与实践记录

《PythonDash框架在数据可视化仪表板中的应用与实践记录》Python的PlotlyDash库提供了一种简便且强大的方式来构建和展示互动式数据仪表板,本篇文章将深入探讨如何使用Dash设计一... 目录python Dash框架在数据可视化仪表板中的应用与实践1. 什么是Plotly Dash?1.1

基于Flask框架添加多个AI模型的API并进行交互

《基于Flask框架添加多个AI模型的API并进行交互》:本文主要介绍如何基于Flask框架开发AI模型API管理系统,允许用户添加、删除不同AI模型的API密钥,感兴趣的可以了解下... 目录1. 概述2. 后端代码说明2.1 依赖库导入2.2 应用初始化2.3 API 存储字典2.4 路由函数2.5 应

Python GUI框架中的PyQt详解

《PythonGUI框架中的PyQt详解》PyQt是Python语言中最强大且广泛应用的GUI框架之一,基于Qt库的Python绑定实现,本文将深入解析PyQt的核心模块,并通过代码示例展示其应用场... 目录一、PyQt核心模块概览二、核心模块详解与示例1. QtCore - 核心基础模块2. QtWid

最新Spring Security实战教程之Spring Security安全框架指南

《最新SpringSecurity实战教程之SpringSecurity安全框架指南》SpringSecurity是Spring生态系统中的核心组件,提供认证、授权和防护机制,以保护应用免受各种安... 目录前言什么是Spring Security?同类框架对比Spring Security典型应用场景传统

Python结合Flask框架构建一个简易的远程控制系统

《Python结合Flask框架构建一个简易的远程控制系统》这篇文章主要为大家详细介绍了如何使用Python与Flask框架构建一个简易的远程控制系统,能够远程执行操作命令(如关机、重启、锁屏等),还... 目录1.概述2.功能使用系统命令执行实时屏幕监控3. BUG修复过程1. Authorization

SpringBoot集成图片验证码框架easy-captcha的详细过程

《SpringBoot集成图片验证码框架easy-captcha的详细过程》本文介绍了如何将Easy-Captcha框架集成到SpringBoot项目中,实现图片验证码功能,Easy-Captcha是... 目录SpringBoot集成图片验证码框架easy-captcha一、引言二、依赖三、代码1. Ea

Gin框架中的GET和POST表单处理的实现

《Gin框架中的GET和POST表单处理的实现》Gin框架提供了简单而强大的机制来处理GET和POST表单提交的数据,通过c.Query、c.PostForm、c.Bind和c.Request.For... 目录一、GET表单处理二、POST表单处理1. 使用c.PostForm获取表单字段:2. 绑定到结

Nginx启动失败:端口80被占用问题的解决方案

《Nginx启动失败:端口80被占用问题的解决方案》在Linux服务器上部署Nginx时,可能会遇到Nginx启动失败的情况,尤其是错误提示bind()to0.0.0.0:80failed,这种问题通... 目录引言问题描述问题分析解决方案1. 检查占用端口 80 的进程使用 netstat 命令使用 ss

Python如何实现PDF隐私信息检测

《Python如何实现PDF隐私信息检测》随着越来越多的个人信息以电子形式存储和传输,确保这些信息的安全至关重要,本文将介绍如何使用Python检测PDF文件中的隐私信息,需要的可以参考下... 目录项目背景技术栈代码解析功能说明运行结php果在当今,数据隐私保护变得尤为重要。随着越来越多的个人信息以电子形

pip install jupyterlab失败的原因问题及探索

《pipinstalljupyterlab失败的原因问题及探索》在学习Yolo模型时,尝试安装JupyterLab但遇到错误,错误提示缺少Rust和Cargo编译环境,因为pywinpty包需要它... 目录背景问题解决方案总结背景最近在学习Yolo模型,然后其中要下载jupyter(有点LSVmu像一个