Data mining

Computer science

Boosting (machine learning)

Pattern recognition (psychology)

Data pre-processing

Artificial intelligence

Deep learning

Class (philosophy)

Convolutional neural network

Machine learning

Preprocessor

Nuclear Engineering and Technology

We describe a new deep learning model - Convolutional eXtreme Gradient Boosting (ConvXGB) for classification problems based on convolutional neural nets and Chen et al.’s XGBoost. As well as image data, ConvXGB also supports the general classification problems, with a data preprocessing module. ConvXGB consists of several stacked convolutional layers to learn the features of the input and is able to learn features automatically, followed by XGBoost in the last layer for predicting the class labels. The ConvXGB model is simplified by reducing the number of parameters under appropriate conditions, since it is not necessary re-adjust the weight values in a back propagation cycle. Experiments on several data sets from UCL Repository, including images and general data sets, showed that our model handled the classification problems, for all the tested data sets, slightly better than CNN and XGBoost alone and was sometimes significantly better.

ConvXGB: A new deep learning model for classification problems based on CNN and XGBoost