New Prompt Learning Boosts Crystal Property Prediction in GNNs

Shrimon Mukherjee, Kishalay Das, Partha Basuchowdhuri, Pawan Goyal, Niloy Ganguly· July 13, 2026 View original

Summary

Researchers developed a lightweight, model-agnostic prompt learning framework for Graph Neural Networks to improve crystal property prediction. This framework captures latent chemical and structural features, significantly enhancing performance and enabling cross-property knowledge transfer.

This research introduces an innovative soft prompt learning framework designed to enhance the accuracy of Graph Neural Networks (GNNs) in predicting crystal properties. Traditional GNNs often struggle with incorporating all relevant chemical and structural features, leading to larger models and reliance on deep domain expertise. The new framework addresses this by capturing latent features not explicitly fed into the GNN. The proposed method employs a multilevel approach, utilizing both node-level prompts to understand local chemical semantics and graph-level prompts to encode global structural symmetry. This lightweight framework integrates seamlessly with existing GNN encoders. Experiments demonstrate a substantial performance improvement, ranging from 3% to 15%, on various benchmark datasets. A key advantage of this prompt learning approach is its ability to facilitate knowledge transfer across different crystal properties. This means that models can achieve better prediction performance even for properties where training data is scarce, making it a versatile tool for materials science.

Why it matters

This advancement offers materials scientists and AI engineers a more efficient and accurate method for predicting crystal properties, accelerating materials discovery and design processes. It reduces the need for extensive domain-specific feature engineering, making GNNs more accessible and powerful.

How to implement this in your domain

  1. 1Integrate the prompt learning framework into existing GNN architectures for materials science applications.
  2. 2Experiment with node-level and graph-level prompts to capture specific chemical and structural nuances in crystal data.
  3. 3Apply the cross-property knowledge transfer capability to improve predictions for novel materials or properties with limited data.
  4. 4Evaluate performance gains on custom crystal datasets to validate the framework's effectiveness in specific research or industrial contexts.

Who benefits

Materials SciencePharmaceuticalsChemical ManufacturingRenewable Energy

Key takeaways

  • A new prompt learning framework significantly improves GNN performance in crystal property prediction.
  • The framework uses both node-level and graph-level prompts to capture latent chemical and structural features.
  • It is lightweight, model-agnostic, and integrates easily with existing GNNs.
  • The approach enables cross-property knowledge transfer, benefiting data-scarce prediction tasks.

Original post by Shrimon Mukherjee, Kishalay Das, Partha Basuchowdhuri, Pawan Goyal, Niloy Ganguly

"arXiv:2607.08996v1 Announce Type: cross Abstract: Graph Neural Networks have emerged as a powerful tool for the fast and accurate prediction of various crystal properties. These models often encode domain-specific knowledge into their graph encoding modules, which increases their…"

View on X

Originally posted by Shrimon Mukherjee, Kishalay Das, Partha Basuchowdhuri, Pawan Goyal, Niloy Ganguly on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research

AI ResearchAI Engineering & DevTools

Data-Efficient Deep Learning Guidelines for Inertial Sensor Classification

This study provides empirical guidelines for estimating the minimum training set size needed for deep learning models in inertial sensor classification tasks. It reveals that accuracy follows a consistent logarithmic growth pattern, allowing for data-efficient planning of recording campaigns and achieving practical stability with fewer samples than traditionally assumed.

Ofir Kruzel, Itzik KlienJul 13, 2026
AI Engineering & DevToolsAI Research

On-Device Adaptive AI Boosts EV Battery Power Prediction

Researchers developed a novel approach for on-device learning in electric vehicles (EVs) that continuously adapts pretrained battery power prediction models to new data. This method significantly improves forecasting performance, reducing mean absolute errors by up to 14.88% with offline adaptation and 7.49% with online adaptation.

Avik Bhatnagar, Anton Paule, Tobias Schuermann, Sebastian Reiter, Oliver BringmannJul 13, 2026
AI ResearchAI Engineering & DevTools

New Differentiable Logic Networks Outperform Fixed-Connection Models

Researchers introduce a novel method for optimizing connections in deep differentiable logic gate networks (LGNs) and lookup table networks (LUTNs), achieving superior performance with significantly fewer gates. The approach allows for parallel learning of optimal gate types and LUT entries, demonstrating improved accuracy on benchmarks like MNIST.

Wout Mommen, Lars Keuninckx, Matthias Hartmann, Werner Van Leekwijck, Piet WambacqJul 13, 2026