|
|
Journal of Zhejiang University SCIENCE A
ISSN 1673-565X(Print), 1862-1775(Online), Monthly
2026 Vol.27 No.3 P.215-230
Predicting permeability coefficients of earth-rock material using an improved generative adversarial network and explainable ensemble learning under small sample conditions
Abstract: Accurate prediction of the permeability coefficient is crucial for evaluating the compaction quality of earthworks. However, during the compaction process, on-site testing is often time-consuming and expensive, leading to fewer samples, which affects prediction accuracy. Moreover, most current predictive models have limited capabilities and tend to be black-box models with poor explainability. To overcome these issues, in this study, we proposed a new method to predict the permeability coefficient of earth-rock material based on an improved generative adversarial network (GAN) and explainable osprey optimization algorithm–Huber loss–light gradient boosting machine (OOA–HL–LightGBM). Firstly, by introducing the Wasserstein distance as the loss function into the conditional generative adversarial network (CGAN), the Wasserstein conditional generative adversarial network (WCGAN) was proposed to generate high-quality data, addressing the issue of insufficient information caused by small samples. Furthermore, by incorporating material and compaction parameters as inputs, a high-accuracy permeability coefficient prediction model was developed using LightGBM with the Huber loss function and the OOA. Finally, the Shapley additive explanation (SHAP) method was introduced into OOA–HL–LightGBM to analyze the specific roles of different features within the dataset to enhance the credibility of the prediction results. The proposed method was applied to a large-scale high-core rockfill dam in southwestern China to thoroughly verify its effectiveness and superiority.
Key words: Permeability coefficient prediction; Light gradient boosting machine (LightGBM); Wasserstein conditional generative adversarial network (WCGAN); Shapley additive explanation (SHAP)
机构:中国农业大学,水利与土木工程学院,中国北京,100083
目的:本文旨在针对土石料渗透系数预测中存在小样本和可解释性不足的问题,探究基于生成模型的数据增强方法,以提高预测模型在小样本条件下的泛化能力。同时,结合具备可解释性的集成学习算法,增强预测结果的可信度,实现对土石料渗透系数的高精度预测。
创新点:1.提出一种基于改进生成对抗网络的数据增强方法,有效提升小样本条件下渗透系数预测模型的性能;2.构建基于改进轻量级梯度提升机(LightGBM)的渗透系数预测模型,结合优化算法实现渗透系数更高精度的预测;3.使用沙普利可加性解释方法(SHAP)对预测结果进行全局和局部解释,增强模型的可解释性;4.将所提方法应用于实际土方工程案例中,验证所提方法在工程实践中的有效性。
方法:1.将瓦瑟斯坦(Wasserstein)距离作为损失函数引入到条件生成对抗网络中,并基于Wasserstein条件生成对抗网络;2.利用LightGBM算法建立具有Huber损失函数和鱼鹰优化算法的高精度渗透系数预测模型;3.使用SHAP方法探究影响预测结果的关键特征,并分析不同特征在数据集中的具体作用。
结论:1.基于Wasserstein条件生成对抗网络的数据增强方法能够生成高质量的样本,有效解决小样本数据问题;2.基于结合Huber损失和鱼鹰优化的LightGBM算法建立的渗透系数预测模型具有较高的预测性能;3.使用SHAP方法能够对预测结果进行全局和局部分析,提升预测模型的可解释性。
关键词组:
References:
Open peer comments: Debate/Discuss/Question/Opinion
<1>
DOI:
10.1631/jzus.A2500127
CLC number:
Download Full Text:
Downloaded:
2044
Download summary:
<Click Here>Downloaded:
5Clicked:
1557
Cited:
0
On-line Access:
2026-03-25
Received:
2025-04-11
Revision Accepted:
2025-07-11
Crosschecked:
2026-03-25