通过融合聚类的潜在空间增强,提升小样本下的分子溶解度预测精度。
Enhancing Drug Discovery: Autoencoder-Based Latent Space Augmentation for Improved Molecular Solubility Prediction using LatMixSol
- 用自编码器压缩分子特征,再按化学相似性分组进行插值生成新数据。
- 在Huuskonen数据集上将模型RMSE降低3.2%至7.6%,R²提升0.5至1.5%。
- 适合数据稀缺的药物发现场景,计算高效且保持化学合理性。
准确预测分子溶解度是早期药物发现的关键,但传统机器学习模型受限于标注数据少和分子描述符高维性。为此,我们提出LatMixSol,一种结合自编码器特征压缩与引导插值的潜在空间增强框架。首先,使用双层自编码器将分子描述符编码至低维潜在空间;随后通过谱聚类对化学结构相似的分子分组,实现组内类似MixUp的插值操作。通过混合簇内成员的潜在向量并解码回原始空间,生成合成样本。在Huuskonen溶解度基准数据集上评估显示,该方法在三种梯度提升回归器(CatBoost、LightGBM、HistGradientBoosting)中均取得稳定提升,RMSE降低3.2%-7.6%,R²提升0.5-1.5%。其中HistGradientBoosting提升最为显著,达7.6%。分析表明,该方法在保持化学有效性的同时扩展了数据多样性,为资源受限的药物发现流程提供了一种高效的建模策略。
原文摘要 · Abstract (English)
Accurate prediction of molecular solubility is a cornerstone of early-stage drug discovery, yet conventional machine learning models face significant challenges due to limited labeled data and the high-dimensional nature of molecular descriptors. To address these issues, we propose LatMixSol, a novel latent space augmentation framework that combines autoencoder-based feature compression with guided interpolation to enrich training data. Our approach first encodes molecular descriptors into a low-dimensional latent space using a two-layer autoencoder. Spectral clustering is then applied to group chemically similar molecules, enabling targeted MixUp-style interpolation within clusters. Synthetic samples are generated by blending latent vectors of cluster members and decoding them back to the original feature space. Evaluated on the Huuskonen solubility benchmark, LatMixSol demonstrates consistent improvements across three of four gradient-boosted regressors (CatBoost, LightGBM, HistGradientBoosting), achieving RMSE reductions of 3.2-7.6% and R-squared increases of 0.5-1.5%. Notably, HistGradientBoosting shows the most significant enhancement with a 7.6% RMSE improvement. Our analysis confirms that cluster-guided latent space augmentation preserves chemical validity while expanding dataset diversity, offering a computationally efficient strategy to enhance predictive models in resource-constrained drug discovery pipelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。