首页 | 官方网站   微博 | 高级检索  
     

基于混合卷积自编码极限学习机的RGB-D物体识别
引用本文:殷云华,李会方.基于混合卷积自编码极限学习机的RGB-D物体识别[J].红外与激光工程,2018,47(2):203008-0203008(8).
作者姓名:殷云华  李会方
作者单位:1.西北工业大学 电子信息学院,陕西 西安 710072;
基金项目:国家自然科学基金青年科学基金(61402368);瞬态冲击技术重点实验室基金(61426060103162606007)
摘    要:有效学习丰富的表征信息在RGB-D目标识别任务中至关重要,是实现高泛化性能的关键。针对卷积神经网络训练时间长的问题,提出了一种混合卷积自编码极限学习机(HCAE-ELM)结构,包括卷积神经网络(CNN)和自编码极限学习机(AE-ELM),该结构合并了CNN的有效性和AE-ELM快速性的优点。它使用卷积层和池化层分别从RGB和深度图来有效提取低阶特征,然后在共享层合并两种模型特征,输入到自编码极限学习机中以得到高层次的特征,最终的特征使用极限学习机(ELM)进行分类,以获得更好的快速泛化能力。文中在标准的RGB-D数据集上进行了评估测试,其实验结果表明,相比较深度学习和其他的ELM方法,文中的混合卷积自编码极限学习机模型取得了良好的测试准确率,并且有效地缩减了训练时间。

关 键 词:极限学习机    卷积神经网络    自编码极限学习机    物体识别
收稿时间:2017-08-05

RGB-D object recognition based on hybrid convolutional auto-encoder extreme learning machine
Affiliation:1.School of Electronics and Information,Northwestern Polytechnical University,Xi'an 710072,China;2.Science and Technology on Transient Impact Laboratory,Beijing 102202,China
Abstract:Learning rich representations efficiently plays an important role in RGB-D object recognition task, which is crucial to achieve high generalization performance. For the long training time of convolutional neural networks, a Hybrid Convolutional Auto-Encoder Extreme Learning Machine Structure (HCAE-ELM) was put forward which included Convolutional Neural Network (CNN) and Auto-Encoder Extreme Learning Machine (AE-ELM), which combined the power of CNN and fast training of AE-ELM. It used convolution layers and pooling layers to effectively abstract lower level features from RGB and depth images separately. And then, the shared layer was developed by combining these features from each modality and fed to an AE-ELM for higher level features. The final abstracted features were fed to an ELM classifier, which led to better generalization performance with faster learning speed. The performance of HCAE-ELM was evaluated on RGB-D object dataset. Experimental results show that the proposed method achieves better testing accuracy with significantly shorter training time in comparison with deep learning methods and other ELM methods.
Keywords:
点击此处可从《红外与激光工程》浏览原始摘要信息
点击此处可从《红外与激光工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号