2024 AIChE Annual Meeting
(62b) Predicting Polymer–Surface Adhesion Strength Via Machine Learning and Transfer Learning
Author
Building upon this foundation, we address a notable challenge: the necessity of extensive datasets for each specific surface to adequately train the surrogate models. Ideally, one should be able to infer information about similar surfaces without having to regenerate a full complement of adhesion data for each new case, making it possible to explore scenarios of insufficient data. To bridge this gap, we demonstrate a transfer learning technique using a deep neural network to improve the accuracy of ML models trained on small datasets by pre-training on a larger database from a related system and fine-tuning the weights of all layers with a small amount of additional data. The shared knowledge from the pre-trained model facilitates the prediction accuracy significantly on small datasets. We also explore the limits of database size on accuracy and the optimal tuning of network architecture and parameters for our learning tasks. While applied to a relatively simple coarse-grained polymer model, the general lessons of this study apply to detailed modeling studies and the broader problems of inverse materials design.