A Novel Multi-Branch Channel Expansion Network for Garbage Image Classification

Due to the lack of data available for training, deep learning hardly performed well in the field of garbage image classification. We choose the TrashNet data set which is widely used in the field of garbage image classification, and try to overcome data deficiencies in this field by optimizing the network structure. In this article, it is found that the deeper network and short-circuit connection, which are generally accepted in the field of deep learning, will not work well on the TrashNet data set. By analyzing and modifying the network structure, we propose an effective method to improve the network performance on TrashNet data set. This method widens the network by expanding branches, and then uses add layers to realize the fusion of feature information. It can make full use of feature information at slight additional computational cost. Using this method to replace the core structure of the Xception network, the performance of the improved network has been improved greatly. Finally, the M-b Xception network proposed by us achieves 94.34% classification accuracy on the TrashNet data set, and has certain advantages over some state-of-the-art methods on multiple indicators. The python code can be download from https://github.com/scp19801980/Trash-classify-M_b-Xception.

[1]  Jason Yosinski,et al.  Deep neural networks are easily fooled: High confidence predictions for unrecognizable images , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2]  Yongxin Yang,et al.  Frankenstein: Learning Deep Face Representations Using Small Data , 2016, IEEE Transactions on Image Processing.

[3]  Zhibin Wang,et al.  AirVis: Visual Analytics of Air Pollution Propagation , 2020, IEEE Transactions on Visualization and Computer Graphics.

[4]  Bo Chen,et al.  Crowd Counting Method Based on Convolutional Neural Network With Global Density Feature , 2019, IEEE Access.

[5]  Dawn Song,et al.  Robust Physical-World Attacks on Deep Learning Models , 2017, 1707.08945.

[6]  Elmer P. Dadios,et al.  Common Garbage Classification Using MobileNet , 2018, 2018 IEEE 10th International Conference on Humanoid, Nanotechnology, Information Technology,Communication and Control, Environment and Management (HNICEM).

[7]  Eduardo A. Soares,et al.  Artificial Intelligence in Automated Sorting in Trash Recycling , 2018, Anais do XV Encontro Nacional de Inteligência Artificial e Computacional (ENIAC 2018).

[8]  Lei Guo,et al.  When Deep Learning Meets Metric Learning: Remote Sensing Image Scene Classification via Learning Discriminative CNNs , 2018, IEEE Transactions on Geoscience and Remote Sensing.

[9]  Mandar Satvilkar.  Image Based Trash Classification using Machine Learning Algorithms for Recyclability Status , 2018 .

[10]  Minghao Yan,et al.  Adaptive Learning Knowledge Networks for Few-Shot Learning , 2019, IEEE Access.

[11]  Yoshua Bengio,et al.  Scaling learning algorithms towards AI , 2007 .

[12]  Marco A. Wehrmeister,et al.  Using Deep Learning and Low-Cost RGB and Thermal Cameras to Detect Pedestrians in Aerial Images Captured by Multirotor UAV , 2018, Sensors.

[13]  Zhuang Wang,et al.  Scaling Up Kernel SVM on Limited Resources: A Low-Rank Linearization Approach , 2012, IEEE Transactions on Neural Networks and Learning Systems.

[14]  Sangjun Kim,et al.  Fast Pedestrian Detection in Surveillance Video Based on Soft Target Training of Shallow Random Forest , 2019, IEEE Access.

[15]  Dong Xu,et al.  Advanced Deep-Learning Techniques for Salient and Category-Specific Object Detection: A Survey , 2018, IEEE Signal Processing Magazine.

[16]  Ananthram Swami,et al.  The Limitations of Deep Learning in Adversarial Settings , 2015, 2016 IEEE European Symposium on Security and Privacy (EuroS&P).

[17]  Sergey Ioffe,et al.  Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning , 2016, AAAI.

[18]  Ulas Bagci,et al.  Lung and Pancreatic Tumor Characterization in the Deep Learning Era: Novel Supervised and Unsupervised Learning Approaches , 2018, IEEE Transactions on Medical Imaging.

[19]  Shihui Ying,et al.  MR Image Super-Resolution via Wide Residual Networks With Fixed Skip Connection , 2019, IEEE Journal of Biomedical and Health Informatics.

[20]  David Zhang,et al.  Deep Cascade Model-Based Face Recognition: When Deep-Layered Learning Meets Small Data , 2020, IEEE Transactions on Image Processing.

[21]  Xiongfei Li,et al.  The OCS-SVM: An Objective-Cost-Sensitive SVM With Sample-Based Misclassification Cost Invariance , 2019, IEEE Access.

[22]  Adeeb Noor,et al.  An Intelligent Learning System Based on Random Search Algorithm and Optimized Random Forest Model for Improved Heart Disease Detection , 2019, IEEE Access.

[23]  Ali Gholipour,et al.  Semi-Supervised Learning With Deep Embedded Clustering for Image Classification and Segmentation , 2019, IEEE Access.

[24]  David Zhang,et al.  F-SVM: Combination of Feature Transformation and SVM Learning via Convex Relaxation , 2015, IEEE Transactions on Neural Networks and Learning Systems.

[25]  David A. Wagner,et al.  Towards Evaluating the Robustness of Neural Networks , 2016, 2017 IEEE Symposium on Security and Privacy (SP).

[26]  Xiaofeng Zhu,et al.  Efficient kNN Classification With Different Numbers of Nearest Neighbors , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[27]  Nikos Komodakis,et al.  Wide Residual Networks , 2016, BMVC.

[28]  Xiangyu Zhang,et al.  ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[29]  Antonio Plaza,et al.  Skip-Connected Covariance Network for Remote Sensing Scene Classification , 2020, IEEE Transactions on Neural Networks and Learning Systems.

[30]  Séverine Dubuisson,et al.  Dynamic Pose-Robust Facial Expression Recognition by Multi-View Pairwise Conditional Random Forests , 2016, IEEE Transactions on Affective Computing.

[31]  Leslie Pack Kaelbling,et al.  Effect of Depth and Width on Local Minima in Deep Learning , 2018, Neural Computation.

[32]  Kang Zhang,et al.  Marginal Deep Architecture: Stacking Feature Learning Modules to Build Deep Learning Models , 2019, IEEE Access.

[33]  Yongjun Xu,et al.  Rethinking the Number of Channels for the Convolutional Neural Network , 2019, ArXiv.

[34]  Minhaz Uddin Ahmed,et al.  Incremental Deep Learning for Robust Object Detection in Unknown Cluttered Environments , 2018, IEEE Access.

[35]  Gary Thung,et al.  Classification of Trash for Recyclability Status , 2016 .

[36]  Wei Luo,et al.  Remote Sensing Image Retrieval Using Convolutional Neural Network Features and Weighted Distance , 2018, IEEE Geoscience and Remote Sensing Letters.

[37]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[38]  Yilin Bei,et al.  Medical Health Big Data Classification Based on KNN Classification Algorithm , 2020, IEEE Access.

[39]  C. Faggio,et al.  Microplastics in the marine environment: Current trends in environmental pollution and mechanisms of toxicological profile. , 2019, Environmental toxicology and pharmacology.

[40]  Yingfeng Cai,et al.  Salient object detection based on multi-scale contrast , 2018, Neural Networks.

[41]  Xiao Huang,et al.  Magnetocardiography-Based Ischemic Heart Disease Detection and Localization Using Machine Learning Methods , 2019, IEEE Transactions on Biomedical Engineering.

[42]  Razvan Pascanu,et al.  On the Number of Linear Regions of Deep Neural Networks , 2014, NIPS.

[43]  Hossein Anvarifar,et al.  Environmental pollution and toxic substances: Cellular apoptosis as a key parameter in a sensible model like fish. , 2018, Aquatic toxicology.

[44]  S. Nagan,et al.  Assessment of Comprehensive Environmental Pollution Index of Kurichi Industrial Cluster, Coimbatore District, Tamil Nadu, India – a Case Study , 2018 .

[45]  Noel E. O'Connor,et al.  Shallow and Deep Convolutional Networks for Saliency Prediction , 2016, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[46]  Junfu Yu,et al.  A Novel Lane-Changing Decision Model for Autonomous Vehicles Based on Deep Autoencoder Network and XGBoost , 2020, IEEE Access.

[47]  Tuan D. Pham,et al.  Geostatistical Simulation of Medical Images for Data Augmentation in Deep Learning , 2019, IEEE Access.

[48]  Guanghui Wang,et al.  Dictionary Representation of Deep Features for Occlusion-Robust Face Recognition , 2019, IEEE Access.

[49]  Naixue Xiong,et al.  A Pedestrian Detection Method Based on Genetic Algorithm for Optimize XGBoost Training Parameters , 2019, IEEE Access.

[50]  François Chollet,et al.  Xception: Deep Learning with Depthwise Separable Convolutions , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[51]  Yang Zhao,et al.  A Stacked Multi-Connection Simple Reducing Net for Brain Tumor Segmentation , 2019, IEEE Access.

[52]  Anton van den Hengel,et al.  Wider or Deeper: Revisiting the ResNet Model for Visual Recognition , 2016, Pattern Recognit..

[53]  Wei Zhao,et al.  Residual convolutional neural network for predicting response of transarterial chemoembolization in hepatocellular carcinoma from CT imaging , 2019, European Radiology.

[54]  Joel Huber.  Dynamic relationships between social norms and pro-environmental behavior: evidence from household recycling , 2020, Behavioural Public Policy.

[55]  Abhishek Das,et al.  Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization , 2016, 2017 IEEE International Conference on Computer Vision (ICCV).

[56]  Qingjie Liu,et al.  Road Extraction by Deep Residual U-Net , 2017, IEEE Geoscience and Remote Sensing Letters.

[57]  Murat Haciomeroglu,et al.  Classification of TrashNet Dataset Based on Deep Learning Models , 2018, 2018 IEEE International Conference on Big Data (Big Data).

[58]  Bo Chen,et al.  MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications , 2017, ArXiv.

[59]  Xiaohui Zhao,et al.  A Comprehensive Study of Smartphone-Based Indoor Activity Recognition via Xgboost , 2019, IEEE Access.

[60]  Qing Fei,et al.  Convolutional Neural Network Based on Extreme Learning Machine for Maritime Ships Recognition in Infrared Images , 2018, Sensors.

[61]  Shawkat K. Guirguis,et al.  Single Image Super Resolution Model Using Learnable Weight Factor in Residual Skip Connection , 2019, IEEE Access.

[62]  Dimitris S. Papailiopoulos,et al.  The Effect of Network Width on the Performance of Large-batch Training , 2018, NeurIPS.

[63]  Jian Sun,et al.  Identity Mappings in Deep Residual Networks , 2016, ECCV.