基于参数共享和知识蒸馏的分子属性预测模型的压缩系统和方法

    公开(公告)号:CN117672412A

    公开(公告)日:2024-03-08

    申请号:CN202410086735.4

    申请日:2024-01-22

    Abstract: 本发明涉及基于参数共享和知识蒸馏的分子属性预测模型的压缩系统和方法。该压缩系统包括:获取模块,其配置为获取分别用于预训练和微调的分子结构数据的训练集、验证集和测试集;参数共享模块,其配置为基于原始的分子属性预测模型来构建参数共享的分子属性预测模型,并对其进行预训练;知识蒸馏模块,其配置为利用预训练的参数共享的分子属性预测模型来构建知识蒸馏的分子属性预测模型,并对其进行预训练;微调模块,其配置为对预训练的知识蒸馏模型进行微调,从而获得最终的分子属性预测模型;输出模块,其配置为将用于微调的分子结构数据的测试集输入至最终的分子属性预测模型,从而输出分子性质预测效果的数据。

Patent Agency Ranking