Manifold Restore Mixing Enhances Protein Representation Learning
Data augmentation improves protein representation learning but often disrupts structural integrity or reduces diversity. The authors identify these structure defects and performance degradation issues in existing methods. They propose Manifold Restore Mixing (MRM) to restore lost structural information while introducing diverse variations. MRM mixes hidden representations of original and augmented data, inspired by manifold mixup techniques. A sample difficulty scheduler adjusts the beta distribution to provide progressively challenging samples during training. Experiments on various backbones and downstream tasks demonstrate the method's effectiveness and generalization. The implementation is available at https://github.com/KingGugu/MRM.