Title: Peeling Context from Cause for Multimodal Molecular Property Prediction
Primary area: learning on graphs and other geometries & topologies
Keywords: Molecular Property Prediction, Causal–Trivial Peeling, Multimodal Representations, Batch-wise Invariance

Abstract: Deep models are used for molecular property prediction, yet they are often hard to interpret and may rely on spurious context rather than causal structure, which degrades reliability under distribution shift and harms predictive performance. We introduce **CLaP**, *Causal Layerwise Peeling*, a framework which separates causal signal from context in a layerwise manner and integrates diverse graph representations of molecules. At each layer, a *causal block* performs a soft split into causal and trivial branches, fuses causal evidence across modalities, and progressively peels batch-coupled context to concentrate on label-relevant structure, thereby limiting shortcut signals and stabilizing layerwise refinement. Across four molecular benchmarks, CLaP reliably improves MAE, MSE, and $R^2$ over competitive baselines. We also obtain atom-level causal saliency maps that highlight substructures responsible for a prediction, providing actionable guidance for targeted molecular edits. Case studies confirm the accuracy of these maps and their alignment with chemical intuition. By peeling context from cause at every layer, the model delivers predictors that are accurate and interpretable for molecular design.