To address the problem in predicting target prices for ship equipment orders where small sample sizes, high feature dimensions, and strong business constraints lead traditional models to overfit and have insufficient generalization ability, a method combining GAN and NVP-D for small-sample data augmentation and price prediction is proposed. This method integrates the advantages of adversarial training in Generative Adversarial Networks (GAN) with the explicit density estimation and stable training characteristics of Normalizing Flow NVP-D. By using dual-weight collaborative optimization of the objective function, it alleviates gradient vanishing and mode collapse, generating high-quality virtual samples that closely follow the real data distribution. Redundant features are removed using Boruta-Lasso joint feature selection to reduce model complexity. CatBoost is employed as the prediction model to complete price estimation. Experiments were conducted on a ship equipment dataset with 33 original samples and 24 features, strictly following the standard procedure of augmentation only within the training set and 5-fold cross-validation. Compared with NVP, NVP-G, MAF, traditional GAN, and Mixup methods, the results show that the proposed integrated model achieves optimal performance when augmenting 400 samples, with an RMSE of 0.0675, MAE of 0.0510, and R2 of 0.9228. After feature selection, prediction accuracy further improves, with RMSE decreasing to 0.0615 and R2 increasing to 0.9341. Limited by the scale of the original samples, the statistical robustness and cross-dataset generalization capability of this method still need validation with larger datasets. However, under the current small-sample constraints, it can effectively alleviate modeling bottlenecks and provide high-precision support for equipment procurement argumentation, budget preparation, and cost control in stages.
Li et al. (Sun,) studied this question.