Vision Language Model is NOT All You Need: Augmentation Strategies for Molecule Language Models

Namkyeong Lee,Siddhartha Laghuvarapu,Chanyoung Park,Jimeng Sun

2024-07-23

Abstract:Recently, there has been a growing interest among researchers in understanding molecules and their textual descriptions through molecule language models (MoLM). However, despite some early promising developments, the advancement of MoLM still trails significantly behind that of vision language models (VLM). This is because unique challenges exist apart from VLM in the field of MoLM due to 1) a limited amount of molecule-text paired data and 2) missing expertise that occurred due to the specialized areas of focus among the experts. To this end, we propose AMOLE, which 1) augments molecule-text pairs with structural similarity preserving loss, and 2) transfers the expertise between the molecules. Specifically, AMOLE enriches molecule-text pairs by sharing descriptions among structurally similar molecules with a novel structural similarity preserving loss. Moreover, we propose an expertise reconstruction loss to transfer knowledge from molecules that have extensive expertise to those with less expertise. Extensive experiments on various downstream tasks demonstrate the superiority of AMOLE in comprehending molecules and their descriptions, highlighting its potential for application in real-world drug discovery. The source code for AMOLE is available at <a class="link-external link-https" href="https://github.com/Namkyeong/AMOLE" rel="external noopener nofollow">this https URL</a>.

Artificial Intelligence

What problem does this paper attempt to address?

### Problems the Paper Aims to Solve

This paper aims to address two unique challenges faced by Molecular Language Models (MoLM):

1. **Insufficient Data**: Compared to Vision Language Models (VLM), the amount of molecular text-pairing data is limited, which restricts the development of MoLM. Specifically, generating molecular text-pairing requires expensive expertise and a significant amount of wet lab time, making it difficult to increase data volume through web crawling.

2. **Lack of Expertise**: Each molecule is typically described by experts from different fields, but due to varying focuses of these experts, many molecules have incomplete descriptions across different experts. For example, in the PubChem database, only about 27K molecules have descriptions from multiple experts, while the remaining 272K molecules have descriptions from only one expert.

To address these issues, the paper proposes the AMOLE method, which includes two main components:
- **Structure Similarity Preservation Loss**: This increases molecular text-pairing data by sharing descriptions among molecules with similar structures and introduces a novel loss function to ensure that the enhanced pairings are more accurately aligned in the representation space.
- **Expertise Transfer Module**: This leverages existing rich descriptions to infer and supplement the missing expertise for other molecules, thereby improving the model's performance in handling molecules with insufficient detailed descriptions.

Through these strategies, AMOLE demonstrates superior performance in various downstream tasks and shows potential application value in real drug discovery processes.