---
title: 'OTMol: Optimal Transport Molecular Alignment'
url: https://www.emergentmind.com/topics/otmol
type: topic
---

# OTMol: Optimal Transport Molecular Alignment

Searching arXiv for OTMol and closely related optimal-transport molecular alignment work.
OTMol is a molecular alignment method that formulates atom-to-atom correspondence and rigid superposition as an optimal transport problem rather than as a heuristic assignment followed by Kabsch alignment. Introduced in “Robust Molecular Structure Comparison via Optimal Transport” [2509.01550], it is designed for structural comparison tasks in which root-mean-square deviation (RMSD) is sensitive to atom ordering, molecular symmetry, indistinguishable cluster configurations, and chirality inversion. Its central technical claim is that meaningful RMSD requires a chemically valid one-to-one correspondence, and that such a correspondence can be obtained more robustly by fused supervised Gromov–Wasserstein optimal transport, which incorporates atom labels together with intra-molecular geometric and topological relations [2509.01550].

## 1. Problem formulation and motivation

RMSD is widely used to assess structural similarity in systems ranging from flexible ligand conformers to complex molecular cluster configurations, but its interpretation depends on the existence of a correct one-to-one atom mapping [2509.01550]. If atom correspondence is incorrect, RMSD may become artificially large or artificially small. OTMol is intended to address precisely this prerequisite step.

The method targets three failure modes identified for conventional RMSD pipelines. First, RMSD depends on atom ordering: the same molecule may be represented with different atom orders by different software, databases, or prediction tools, and a wrong correspondence invalidates the comparison [2509.01550]. Second, symmetric or indistinguishable cluster configurations, such as water clusters or noble gas clusters, admit many nearly equivalent permutations; a valid method must avoid splitting one molecule across several others or assigning atoms from one water molecule into multiple different water molecules [2509.01550]. Third, chirality inversion can arise when reflections are allowed during rigid alignment, yielding a smaller RMSD for a mirrored but chemically invalid structure; the paper identifies this as a major defect of ArbAlign-style procedures [2509.01550].

These issues motivate OTMol’s reformulation of molecular alignment as an optimal transport problem. Rather than treating atom assignment as an isolated combinatorial step driven by a hand-designed inter-atom cost matrix, OTMol uses the molecule’s own internal geometry and topology to determine correspondence [2509.01550]. This suggests a shift from “atom similarity” in the narrow sense to relation-preserving matching across whole structures.

## 2. Mathematical framework: from OT to fused supervised Gromov–Wasserstein

OTMol is built on Gromov–Wasserstein (GW) optimal transport. In the paper’s formulation, standard Kantorovich OT uses a cost matrix \(C_{ij}\) between source and target atoms, whereas GW matches objects by preserving relations within each object rather than by comparing atoms directly across objects [2509.01550]. For molecules, this is natural because molecular identity is encoded not only by atom labels but by local geometry and topology.

The paper states that GW seeks
$$
\mathop{\arg \min}_{\mathbf P\in \Gamma} \sum_{i,j,k,l} (\mathbf D^{\mathrm{A}}_{ik} - \mathbf D^{\mathrm{B}}_{jl})^2 \mathbf P_{ij}\mathbf P_{kl}
$$
with
$$
\Gamma = \left\{\mathbf P \in \mathbb R_+^{n\times n} : \mathbf P\mathbf 1_n = \frac1n\mathbf 1_n,\; \mathbf P^T\mathbf 1_n = \frac1n\mathbf 1_n \right\}.
$$
Here \(\mathbf P_{ij}\) is the probability that atom \(i\) in molecule A matches atom \(j\) in molecule B, and the objective penalizes mappings that fail to preserve pairwise distances across the two structures [2509.01550].

OTMol does not rely on GW alone. Its core formulation is fused supervised Gromov–Wasserstein (fsGW), which combines a supervised OT term enforcing label consistency with a GW term preserving structural geometry [2509.01550]:
$$
\mathop{\arg \min}_{\mathbf P\in \Gamma}\ (1-\alpha)\langle \mathbf C, \mathbf P\rangle_F + \alpha \sum_{i,j,k,l}(\mathbf D^{\mathrm{A}}_{ik} - \mathbf D^{\mathrm{B}}_{jl})^2 \mathbf P_{ij}\mathbf P_{kl}.
$$
The parameter \(\alpha \in [0,1]\) controls the balance between label matching and structure matching. The supervised cost matrix satisfies
$$
\mathbf C_{ij}=0 \text{ if labels match, } \infty \text{ otherwise,}
$$
so chemically forbidden assignments are excluded [2509.01550]. The distance matrix entering the GW term is defined as
$$
\mathbf D = (1-c)\mathbf D_E + c\mathbf D_G, \qquad c\in[0,1],
$$
where \(\mathbf D_E\) is the Euclidean distance matrix and \(\mathbf D_G\) is the graph geodesic distance matrix [2509.01550]. Euclidean distances preserve 3D shape, whereas geodesic distances preserve bond and connectivity structure.

This formulation gives OTMol its “chemistry-aware” character in the paper’s sense: element or atom-type consistency can be imposed by \(\mathbf C\), while internal geometry and bond topology are preserved by the GW component [2509.01550]. A plausible implication is that the method is less dependent on local hand-crafted atom costs than Hungarian-style heuristics.

## 3. Construction of atom mappings and rigid alignment

The transport plan \(\mathbf P\) produced by fsGW is a probability matrix rather than a permutation. OTMol therefore includes an explicit projection step that converts transport into a hard assignment. The paper defines
$$
\mathbf P'_{ij} = 1 \quad \text{when } j = \arg\max_k \mathbf P_{ik},
$$
and then checks whether \(\mathbf P'\) is a valid permutation matrix [2509.01550]. This validity test is essential because RMSD requires a one-to-one mapping and because it prevents many-to-one collapse.

Once a valid permutation is obtained, OTMol applies the Kabsch algorithm to compute the optimal rigid-body superposition and then the RMSD [2509.01550]. Chirality preservation is enforced by using Kabsch without reflection, so only proper rotations are allowed. The paper emphasizes that this is crucial for ATP and the FGG peptide, where reflected alignments can reduce RMSD while inverting stereochemistry [2509.01550].

For single molecules with the same number of atoms, the pipeline is specified as follows [2509.01550]. The inputs are coordinates \(X^{\mathrm A}, X^{\mathrm B}\), atom labels \(T^{\mathrm A}, T^{\mathrm B}\), and optional adjacency matrices \(B^{\mathrm A}, B^{\mathrm B}\). The cost matrix is constructed by label equality, the distance matrices are formed using \(\mathbf D = (1-c)\mathbf D_E + c\mathbf D_G\), and the method sweeps over
$$
L_\alpha=\{0,0.01,\dots,1\}.
$$
For each \(\alpha\), fsGW is solved, \(\mathbf P\) is projected to \(\mathbf P'\), invalid assignments are discarded, Kabsch is applied, and the best result is selected by lowest bond connectivity inconsistency (BCI) first and then lowest RMSD [2509.01550].

The prioritization of BCI reflects a substantive methodological choice. The paper states that the mapping should preserve bonds: a bond in the reference should still be a bond after mapping [2509.01550]. This makes the accepted correspondence close to a graph isomorphism rather than merely an RMSD-minimizing permutation.

## 4. Specialized variants for clusters and highly symmetric systems

A notable feature of OTMol is that it does not use a single undifferentiated procedure for all molecular systems. The paper distinguishes among single molecules, noble gas clusters, and clusters of the same molecule, especially water clusters [2509.01550].

For noble gas clusters, label information is absent, so the supervised cost matrix is unnecessary. The method builds the Euclidean distance matrix, optionally raises it elementwise to a Hadamard power \(p\), solves GW, converts the transport plan to a permutation, applies Kabsch, and then refines the result by solving a Kantorovich OT problem on the post-alignment distance matrix:
$$
\mathop{\arg \min}_{\mathbf P \in \Gamma}\langle \mathbf P,\mathbf M\rangle_F,
$$
where
$$
\mathbf M_{ij}=\|\mathbf x_i^{\mathrm A}-\mathbf T(\mathbf x_j^{\mathrm B})\|_2^2.
$$
The paper describes this as a two-stage procedure that is particularly useful for highly symmetric systems [2509.01550].

For clusters of the same molecule, especially water clusters, OTMol introduces a hierarchical alignment strategy because molecule-level integrity must be preserved. The problem is that atoms belonging to one water molecule must not be mapped across several different waters [2509.01550]. The solution is to represent each molecule by a centroid or other representative coordinate, add perturbations to these representative coordinates to generate multiple candidate molecule-level assignments, solve GW or OT at the representative level, and then restrict atom-level matching to the assigned molecular partner by a block cost matrix [2509.01550]. Kabsch is then applied, RMSD is computed, and the process is repeated over many perturbations.

The paper notes that the optimal atom-level correspondence may not arise from the optimal molecule-level correspondence, so multiple suboptimal molecule-level matchings are sampled by adding Gaussian noise before GW [2509.01550]. This is especially important in symmetric systems such as water clusters, where naïve methods can achieve deceptively good RMSD values while producing mismatched water molecules.

## 5. Chemical constraints: chirality, bond consistency, and one-to-one molecular integrity

OTMol’s defining claim is not merely that it lowers RMSD, but that it does so while preserving chemically meaningful structure [2509.01550]. Three constraints are emphasized.

The first is chirality preservation. By using Kabsch without reflection, OTMol prevents mirror inversions that would flip stereocenters [2509.01550]. The paper contrasts this with ArbAlign-style procedures, where reflections among candidate rigid transforms may lead to chemically wrong but numerically favorable alignments.

The second is bond connectivity consistency. The use of graph geodesic distances \(\mathbf D_G\) and BCI-based selection is intended to ensure that if two atoms are bonded in the reference, their mapped counterparts remain bonded after alignment [2509.01550]. This is particularly important for peptides, sugars, lipids, DNA, and atmospheric clusters. The paper’s discussion treats BCI as a structural filter rather than as a secondary descriptive statistic.

The third is one-to-one molecular integrity in cluster comparison. For molecular clusters, OTMol enforces one-to-one mappings between entire molecules, thereby avoiding erroneous many-to-one alignments that can arise when equivalent monomers are present [2509.01550]. In water clusters, this prevents a water molecule’s oxygen and hydrogens from being distributed across multiple waters [2509.01550].

These constraints distinguish OTMol from approaches that optimize only a geometric objective. The paper argues that lower RMSD alone can be misleading if achieved through bond-breaking mappings, high atom mismatch inconsistency (AMI), or chirality inversion [2509.01550].

## 6. Empirical evaluation across molecular systems

The paper evaluates OTMol on a broad set of systems that includes biomedical molecules, cyclic peptides, lipids, sugars, DNA, water clusters, atmospheric hydrates, and noble gas clusters [2509.01550]. The listed systems include ATP, Imatinib, cyclic peptides, phospholipid DLP, unsaturated fatty acid EIC, sugar systems BGC and BGC_GLC, DNA (215D), water clusters, FGG tripeptides, S1-MA-W1 atmospheric hydrate, and neon clusters [2509.01550].

The reported quantitative findings are summarized below.

| System category | Reported OTMol result | Reported comparison |
|---|---|---|
| Biomedical molecules | \(0\%\) BCI on all aligned biomedical examples | ArbAlign often had substantial bond mismatches |
| ATP | OTMol preserved molecular integrity | ArbAlign produced mappings with mean BCI around \(28.46\%\), \(35.48\%\) AMI, and chirality inversion in some cases |
| Imatinib | OTMol preserved molecular integrity | ArbAlign had \(21.23\%\) BCI and \(81.08\%\) AMI in the reported example |
| DLP | OTMol preserved molecular integrity | ArbAlign mean BCI around \(49.74\%\) |
| BGC | OTMol preserved molecular integrity | ArbAlign mean BCI around \(16.80\%\) |
| BGC_GLC | OTMol preserved molecular integrity | ArbAlign mean BCI around \(30.20\%\) |
| EIC | OTMol preserved molecular integrity | ArbAlign mean BCI around \(24.74\%\) |
| DNA | OTMol preserved molecular integrity | ArbAlign had \(0\%\) BCI in that table |
| Similar-dihedral cyclic peptides | RMSD below \(0.5\ \text{\AA}\) for all pairs | \(0\%\) mismatched bond percentage |
| Water clusters | Preserved one-to-one molecular assignments | ArbAlign could obtain lower RMSD in some cases, but often by producing mismatched water molecules |
| Neon clusters | Aligned all pairs closely except one 200-atom case | Scaled much better than ArbAlign for larger systems |

For biomedical molecules, OTMol achieved \(0\%\) BCI on all aligned biomedical examples, meaning bond connectivity was fully preserved [2509.01550]. In contrast, ArbAlign often exhibited substantial bond mismatches: ATP around \(28.46\%\), Imatinib \(21.23\%\), DLP \(49.74\%\), BGC \(16.80\%\), BGC_GLC \(30.20\%\), EIC \(24.74\%\), and DNA \(0\%\) in that table [2509.01550]. In ATP, ArbAlign also produced mappings with \(35.48\%\) AMI and chirality inversion in some cases; in Imatinib, the reported example had \(81.08\%\) AMI [2509.01550].

For cyclic peptides, OTMol gave RMSD below \(0.5\ \text{\AA}\) for all pairs in the similar-dihedral set and preserved all bonds, corresponding to \(0\%\) mismatched bond percentage [2509.01550]. For water clusters, the paper reports that OTMol generally outperformed ArbAlign in preserving one-to-one molecular assignments, even though ArbAlign could sometimes obtain a lower RMSD by producing mismatched water molecules [2509.01550]. For neon clusters, OTMol aligned all pairs closely except one 200-atom case and scaled much better than ArbAlign for larger systems [2509.01550].

The paper also compares against ArbAlign EMD, where the Hungarian solver is replaced by Earth Mover’s Distance from POT. This variant is still generally worse than OTMol, which the authors interpret as evidence that changing the solver does not address the deeper limitation imposed by heuristic atom-cost design [2509.01550].

## 7. Computational properties, sensitivity analyses, and limitations

All three OTMol variants are stated to have theoretical time complexity \(O(n^3)\) [2509.01550]. The paper reports that OTMol is much faster than ArbAlign for large systems: ArbAlign runtime increases steeply beyond 100 atoms, whereas OTMol remains more scalable [2509.01550]. This combination of cubic worst-case complexity and better empirical scalability is important to the method’s positioning as a generalizable structural comparison framework.

The paper includes several sensitivity studies. OTMol sweeps over \(\alpha\in\{0,0.01,\dots,1\}\), and for some systems different \(\alpha\) values trade off label fidelity and geometry fidelity; in FGG, a higher-quality alignment was obtained at \(\alpha=0.62\) than at \(\alpha=0.01\) [2509.01550]. The parameter \(c\), which mixes Euclidean and geodesic distances, is often set to \(c=1\) for strong bond/connectivity preservation in biomolecules and \(c=0.5\) for cyclic peptides and FGG [2509.01550]. For noble gas clusters, multiple Hadamard powers \(p\) are tested, and RMSD does not always improve monotonically with increasing \(p\) [2509.01550]. For water clusters, increasing the perturbation count \(l\) helps find better molecule-level mappings; for 10-PP1 and 10-PP2, RMSD decreased as \(l\) increased [2509.01550].

The method also has explicit assumptions and limitations. It assumes that the two systems are comparable under a one-to-one matching model at the relevant level, often assumes equal-size structures for the main single-molecule formulation, and depends on meaningful atom labels or graph structure when fsGW is used [2509.01550]. The permutation extraction step is heuristic, based on row-wise argmax followed by a permutation check [2509.01550]. The transport solution is not always exactly a permutation, so projection is necessary; the hyperparameters \(\alpha\), \(c\), \(p\), and perturbation count \(l\) affect performance; and for highly symmetric clusters, multiple near-equivalent solutions may exist, making the final result sampling-dependent [2509.01550]. The paper further notes that while OTMol scales better for large systems and is more chemically reliable, it can be more computationally intensive than a naïve single Hungarian solve for tiny systems [2509.01550].

Within molecular modeling and cheminformatics, OTMol is therefore best understood as a method for robust structural correspondence under geometric, topological, and chemical constraints. Its contribution is not the replacement of RMSD, but the replacement of brittle atom matching by a fused optimal transport formulation that is independent of atom ordering, robust to symmetry and indistinguishable cluster configurations, and chirality-preserving [2509.01550].

Source: https://www.emergentmind.com/topics/otmol