Image Semantic Segmentation Scheme based on XGBoost combination with Convolution Feature Extraction

被引:1
作者
Dai, Zichen [1 ]
Liu, Xuewen [1 ]
Xu, Chi [1 ,2 ,3 ]
机构
[1] China Univ Geosci, Sch Automat, Wuhan 430074, Peoples R China
[2] Hubei Key Lab Adv Control & Intelligent Automat C, Wuhan 430074, Peoples R China
[3] Minist Educ, Engn Res Ctr Intelligent Technol Geoexplorat, Wuhan 430074, Peoples R China
来源
2022 41ST CHINESE CONTROL CONFERENCE (CCC) | 2022年
关键词
Image Semantic Segmentation; Convolutional Feature Extraction; XGBoost; Gaussian Filtering; Low Training Cost; NEURAL-NETWORK;
D O I
10.23919/CCC55666.2022.9901915
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Aiming at the problem that the existing image semantic segmentation schemes based on convolutional neural network require a large number of training data sets and high training cost, this paper presents a novel scheme for image semantic segmentation based on extreme gradient boosting (XGBoost), a scalable tree boosting system and the combination of convolution feature extraction (CFE). The scheme consists of three stages. First, convolution operation is performed on a small amount of RGB pixel information of the image processed by Gaussian filtering to acquire more multi-dimensional image feature information. Then, some preprocessing techniques of image feature information, including feature normalization and image label pixel number uniformity are performed to match the model. Finally, the preprocessed image feature information and corresponding labels are input into XGBoost to train and predict the data of the test set. Through simulation and practical tests, compared with other commonly used few data classification algorithms like k-Nearest-Neighbor (kNN) and Support-Vector-Machine (SVM), this scheme can achieve high precision pixel-level image semantic segmentation under the condition of lower training cost.
引用
收藏
页码:6334 / 6340
页数:7
相关论文
共 35 条
[1]  
Albawi S, 2017, I C ENG TECHNOL
[2]   Recurrent residual U-Net for medical image segmentation [J].
Alom, Md Zahangir ;
Yakopcic, Chris ;
Hasan, Mahmudul ;
Taha, Tarek M. ;
Asari, Vijayan K. .
JOURNAL OF MEDICAL IMAGING, 2019, 6 (01)
[3]  
[Anonymous], P 22 ACM SIGKDD INT
[4]  
[Anonymous], J PHYS C SERIES
[5]  
[Anonymous], J PERS MED
[6]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[7]   A Novel PCA-Firefly Based XGBoost Classification Model for Intrusion Detection in Networks Using GPU [J].
Bhattacharya, Sweta ;
Krishnan, Siva Rama S. ;
Maddikunta, Praveen Kumar Reddy ;
Kaluri, Rajesh ;
Singh, Saurabh ;
Gadekallu, Thippa Reddy ;
Alazab, Mamoun ;
Tariq, Usman .
ELECTRONICS, 2020, 9 (02)
[8]   A novel systematic and evolved approach based on XGBoost-firefly algorithm to predict Young's modulus and unconfined compressive strength of rock [J].
Cao, Jing ;
Gao, Juncheng ;
Rad, Hima Nikafshan ;
Mohammed, Ahmed Salih ;
Hasanipanah, Mahdi ;
Zhou, Jian .
ENGINEERING WITH COMPUTERS, 2022, 38 (SUPPL 5) :3829-3845
[9]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[10]  
Chengcheng G., 2018, PROGR LASER OPTOELEC, V55, P235