利用梯度提升树分类法实现乳腺癌数据集分类

目录

• 1. 作者介绍
• 2. 梯度提升树算法
• • 2.1 Boosting 算法
  • 2.2 Boosting Tree （提升树）
  • 2.3 梯度提升树（Gradient Boosting Tree）
• 3. 利用梯度提升树分类法实现乳腺癌数据集分类实验
• • 3.1 乳腺癌数据集介绍
  • 3.2 实验过程
  • 3.3 实验结果
  • 3.4 完整代码
• 4. 参考文献

1. 作者介绍

谈翠红，女，西安工程大学电子信息学院，2023级研究生
研究方向：机器视觉与人工智能
电子邮件：t19856597379@163.com

徐达，男，西安工程大学电子信息学院，2023级研究生，张宏伟人工智能课题组
研究方向：机器视觉与人工智能
电子邮件：1374455905@qq.com

2. 梯度提升树算法

2.1 Boosting 算法

从图可以看出，Boosting 算法的工作机制是从训练集用初始权重训练出一个弱学习器1，根据弱学习器的学习误差率来更新训练样本的权重，使得之前弱学习器1中学习误差率高的训练样本点权重变高。然后这些误差率高的点在弱学习器2中得到更高的重视，利用调整权重后的训练集来训练弱学习器2。如此重复进行，直到弱学习器数达到事先指定的数目T，最终将这T个弱学习器通过集合策略进行整合，得到最终的强学习器。了解Boosting方法后，我们便可将Boosting方法和Decision Tree相结合便可得到Boosting Decision Tree。

2.2 Boosting Tree （提升树）

提升树是以决策树为基本学习器的提升方法，它被认为是统计学习中性能最好的方法之一。对于分类问题，提升树的决策树是二叉决策树，对于回归问题，提升树中的决策是二叉回归树。不同问题的提升树学习算法主要区别在于使用的损失函数不同。
提升树模型可以表示为决策树为基学习器的加法模型：

其中，hm(x ;θm)表示第m个决策树，θm为第 m个决策树的参数，M为决策树的数量。

从图能够直观的看到，提升树的学习思想有点类似打高尔夫球，先粗略的打一杆，然后在之前的基础上逐步靠近球洞，也就是说每一棵树学习的是之前所有树的结论和残差，这个残差就是一个加预测值后得到真实值的累加量。

2.3 梯度提升树（Gradient Boosting Tree）

基于前面介绍的提升树基本思路，可以发现这个算法没有解决损失函数拟合方法的问题。针对这个问题，Freidman提出用损失函数的负梯度来拟合损失的近似值，所以结合得到了梯度提升树算法。而本次实验所使用的是梯度提升树分类算法，它是通过逐步构建一系列的决策树，每棵树都拟合前一棵树的残差，从而逐步提高模型的预测性能。通过对乳腺癌数据集的分类，梯度提升树能够有效地区分良性和恶性肿瘤，同时提供每个特征的重要性，从而帮助理解哪些特征对预测结果影响最大。它的核心原理如下：
初始化一个弱学习器F0(x) ，通常使用一个简单的初始模型（如平均值）来拟合目标变量y 。
对于m=1, 2, . . . , M （M 为迭代次数），迭代进行以下步骤：
计算当前模型的残差rim = yi-F(m-1)(xi) ，其中 F(m-1)(xi) 是前 m-1 个模型的组合预测结果。
拟合一个新的基本学习器 hm(x) ，使得 hm(x) 在训练集上拟合残差 rim 。
更行模型： Fm(x) = F(m-1)(x) +α hm(x) ，其中是一个学习率（也称为步长），控制每次迭代新模型的贡献程度。
最终模型为FM(x) 。

3. 利用梯度提升树分类法实现乳腺癌数据集分类实验

3.1 乳腺癌数据集介绍

威斯康星州乳腺癌数据集是scikit-learn（sklearn）库中一个常用的内置数据集，用于分类任务。该数据集包含了从乳腺癌患者收集的肿瘤特征的测量值，以及相应的良性（benign）或恶性（malignant）标签。以下是对该数据集的简单介绍：
数据集来源：数据集最初由威斯康星州医院的Dr. William H. Wolberg收集。
数据集大小：569 个样本，其中良性样本357个，恶性样本212个
特征数量：30 个数值特征
特征名称：每个特征表示从乳腺细胞核图像中提取的一个属性。以下是特征的具体描述：
• 半径（mean radius）
• 纹理（mean texture）
• 周长（mean perimeter）
• 面积（mean area）
• 平滑度（mean smoothness）
• 紧致度（mean compactness）
• 凹陷点（mean concavity）
• 凹点数（mean concave points）
• 对称性（mean symmetry）
• 分形维数（mean fractal dimension）
以上特征的计算方式包括平均值（mean）、标准误（standard error）、最大值（worst），每个特征的这三种计算方式形成了30个特征。

3.2 实验过程

下面展示一些 内联代码片。

pip install numpy pandas scikit-learn matplotlib seaborn -i 
https://pypi.tuna.tsinghua.edu.cn/simple

numpy: 用于处理数值计算
pandas: 用于数据处理和分析
scikit-learn: 加载乳腺癌数据集
Matplotlib：提供基础的绘图功能，如线条图、散点图、柱状图等，支持广泛的图表类型和自定义，可以创建静态图形以及一些动态和交互式图形。
Seaborn：基于Matplotlib，提供更高级的数据可视化功能，如时间序列数据的静态图表，自动图形美化，特别适合统计数据可视化
1 首先导入相关库

# 导入库
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.preprocessing import StandardScaler
from sklearn.ensemble import GradientBoostingClassifier
from sklearn.metrics import classification_report, confusion_matrix, accuracy_score
import seaborn as sns

2 数据加载和预处理，将特征数据存储在 DataFrame 中，标签数据存储在 Series 中，使用StandardScaler 对特征数据进行标准化处理，以确保特征值在同一量级。

# 导入库
data = load_breast_cancer()
X = pd.DataFrame(data.data, columns=data.feature_names)
y = pd.Series(data.target)
scaler = StandardScaler()
X_scaled = scaler.fit_transform(X)

3 将数据集划分为训练集和测试集 训练集70% 、测试集30%

X_train, X_test, y_train, y_test = train_test_split(X_scaled, y, test_size=0.3, random_state=42)

4 使用 Gradient Boosting Classifier 进行模型训练

gbc = GradientBoostingClassifier(random_state=42)
gbc.fit(X_train, y_train)

5 使用测试集进行模型评估

y_pred = gbc.predict(X_test)
print("初始模型评估:")
print(classification_report(y_test, y_pred))
print("混淆矩阵:")
print(confusion_matrix(y_test, y_pred))
print("准确率:", accuracy_score(y_test, y_pred))

6优化参数：网格搜索使用 5 折交叉验证评估每个超参数组合，总共有 64 种超参数组合，因此需要进行 320 次模型训练和验证，得到最佳参数。

param_grid = {
    'n_estimators': [50, 100, 150, 200],
    'learning_rate': [0.01, 0.05, 0.1, 0.2],
    'max_depth': [3, 4, 5, 6]
}
rid_search = GridSearchCV(estimator=gbc, param_grid=param_grid, cv=5, n_jobs=1, verbose=2)
grid_search.fit(X_train, y_train)
print("最佳参数:", grid_search.best_params_)

7 优化后的模型评估

best_gbc = grid_search.best_estimator_
y_pred_best = best_gbc.predict(X_test)
print("优化后模型评估:")
print(classification_report(y_test, y_pred_best))
print("混淆矩阵:")
print(confusion_matrix(y_test, y_pred_best))
print("准确率:", accuracy_score(y_test, y_pred_best))

8 混淆矩阵可视化

conf_matrix = confusion_matrix(y_test, y_pred_best)
plt.figure(figsize=(10, 7))
sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', xticklabels=data.target_names, yticklabels=data.target_names)
plt.xlabel('Predicted')
plt.ylabel('True')
plt.title('Confusion Matrix')
plt.show()

9 特征重要性可视化

feature_importances = best_gbc.feature_importances_
features = data.feature_names
indices = np.argsort(feature_importances)
plt.figure(figsize=(15, 10))
plt.title('Feature Importances')
plt.barh(range(len(indices)), feature_importances[indices], color='b', align='center')
plt.yticks(range(len(indices)), [features[i] for i in indices])
plt.xlabel('Relative Importance')
plt.show()

3.3 实验结果

得到最佳参数：学习率0.05、决策树的最大深度3、决策树的数量200
模型分类准确率为96%，如果计算机内存够大，可通过设置参数n_jobs=2/3/4得到更高准确率。

混淆矩阵可视化图
0 代表恶性（Malignant）1 代表良性（Benign）
59：表示模型将真实为类别0的样本预测为类别0的数量
105：表示模型将真实为类别1的样本预测为类别1的数量
3：表示模型将真实为类别1的样本错误地预测为类别0的数量
4：表示模型将真实为类别0的样本错误地预测为类别1的数量

特征重要性图
这些特征的重要性分布揭示了在乳腺癌分类问题中，某些特征（如半径、周长和面积）比其他特征（如平滑度、对称性）更有影响力。理解这些特征的重要性有助于进一步改进模型，并且在实际应用中可能提供更多的临床意义

3.4 完整代码

# 导入库
import numpy as np    # 处理数值计算
import pandas as pd    # 数据处理和分析
from sklearn.datasets import load_breast_cancer   # 加载乳腺癌数据集
# 将数据集划分为训练集和测试集以及超参数搜索和交叉验证
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.preprocessing import StandardScaler   # 数据标准化
from sklearn.ensemble import GradientBoostingClassifier   # 梯度提升树分类器模型
# 生成分类报告、混淆矩阵，计算准确率
from sklearn.metrics import classification_report, confusion_matrix, accuracy_score
# 绘制数据可视化图表
import seaborn as sns
import matplotlib.pyplot as plt
# 加载乳腺癌数据集
data = load_breast_cancer()
# 将特征数据存储在 DataFrame 中，标签数据存储在 Series 中
X = pd.DataFrame(data.data, columns=data.feature_names)
y = pd.Series(data.target)
# 数据预处理
scaler = StandardScaler()   # 标准化特征值，使每个特征具有相同的尺度
X_scaled = scaler.fit_transform(X)
# 数据分割 将数据集划分为训练集和测试集 训练集70%  测试集30%
X_train, X_test, y_train, y_test = train_test_split(X_scaled, y, test_size=0.3, random_state=42)
# 使用梯度提升树分类器模型训练
gbc = GradientBoostingClassifier(random_state=42)
gbc.fit(X_train, y_train)
# 使用测试集进行模型评估
y_pred = gbc.predict(X_test)
print("初始模型评估:")
print(classification_report(y_test, y_pred))
print("混淆矩阵:")
print(confusion_matrix(y_test, y_pred))
print("准确率:", accuracy_score(y_test, y_pred))
# 模型优化  决策树的数量  学习率   决策树的最大深度
param_grid = {
'n_estimators': [50, 100, 150, 200],
'learning_rate': [0.01, 0.05, 0.1, 0.2],
'max_depth': [3, 4, 5, 6]
}
grid_search = GridSearchCV(estimator=gbc, param_grid=param_grid, cv=5, n_jobs=1, verbose=2)
grid_search.fit(X_train, y_train)
print("最佳参数:", grid_search.best_params_)
# 优化后的模型评估
best_gbc = grid_search.best_estimator_
y_pred_best = best_gbc.predict(X_test)
print("优化后模型评估:")
print(classification_report(y_test, y_pred_best))
print("混淆矩阵:")
print(confusion_matrix(y_test, y_pred_best))
print("准确率:", accuracy_score(y_test, y_pred_best))
# 混淆矩阵可视化
conf_matrix = confusion_matrix(y_test, y_pred_best)
plt.figure(figsize=(10, 7))
sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', xticklabels=data.target_names, yticklabels=data.target_names)
plt.xlabel('Predicted')
plt.ylabel('True')
plt.title('Confusion Matrix')
plt.show()
# 特征重要性可视化
feature_importances = best_gbc.feature_importances_ # 包含了每个特征的相对重要性
features = data.feature_names  # 所有特征名称
# 返回一个数组，数组元素按特征重要性从小到大的顺序排列
indices = np.argsort(feature_importances)
# 绘制特征重要性图
plt.figure(figsize=(15, 10))
plt.title('Feature Importances')
plt.barh(range(len(indices)), feature_importances[indices], color='b', align='center')
plt.yticks(range(len(indices)), [features[i] for i in indices])
plt.xlabel('Relative Importance')
plt.show()

4. 参考文献

机器学习之梯度提升决策树(GBDT)-腾讯云开发者社区-腾讯云 (tencent.com): link
GBT、GBDT、GBRT与Xgboost – 别再闹了 – 博客园 (cnblogs.com): link