这是一个用于机械故障检测的项目。之所以叫Sparrow的原因,可能是因为鸟类的视力特别好,可以发现很多"Bugs"。再者说,他们也很可爱。在当前版本,我们在实验室测试了一些行星齿轮箱的振动数据,总共有8个错误类型。并使用胶囊神经网络用于模式识别,在振动中检测错误类型,其表现非常不错。
- 我们在实验中获得了一些振动数据
- 并且我们有足够的理由相信机械内部结构损坏可以表现在振动频率上。而利用离散傅里叶变换或者短时傅里叶变换就可以得到信号的频率信息
- 明显,我们可以将一个一维数据变为一张图,输入至神经网络中
- 明显,Hinton提出的胶囊神经网络具有良好的表现性能。因此我们选择使用该架构处理数据
其训练过程中正确率与损失变化曲线如下
如何在高噪声环境下依然具有较高的故障检测正确率,这是一个问题。使用小波变换可能会是一种方法
git clone https://github.com/ecstayalive/Sparrow
git submodule --init --recursive
python main.py
@article{sabour2017dynamic,
title={Dynamic Routing Between Capsules},
author={Sabour, Sara and Frosst, Nicholas and Hinton, Geoffrey E},
journal={arXiv preprint arXiv:1710.09829},
year={2017}
}