Transformer-based Cross-Modal Recipe Embeddings with Large Batch Training
About
In this paper, we present a cross-modal recipe retrieval framework, Transformer-based Network for Large Batch Training (TNLBT), which is inspired by ACME~(Adversarial Cross-Modal Embedding) and H-T~(Hierarchical Transformer). TNLBT aims to accomplish retrieval tasks while generating images from recipe embeddings. We apply the Hierarchical Transformer-based recipe text encoder, the Vision Transformer~(ViT)-based recipe image encoder, and an adversarial network architecture to enable better cross-modal embedding learning for recipe texts and images. In addition, we use self-supervised learning to exploit the rich information in the recipe texts having no corresponding images. Since contrastive learning could benefit from a larger batch size according to the recent literature on self-supervised learning, we adopt a large batch size during training and have validated its effectiveness. In the experiments, the proposed framework significantly outperformed the current state-of-the-art frameworks in both cross-modal recipe retrieval and image generation tasks on the benchmark Recipe1M. This is the first work which confirmed the effectiveness of large batch training on cross-modal recipe embeddings.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image-to-recipe retrieval | Recipe1M 10k setup (test) | Recall@156.5 | 125 | |
| Recipe-to-image retrieval | Recipe1M 10k setup (test) | R@155.9 | 120 | |
| Image-to-recipe retrieval | Recipe1M 1k setup (test) | Recall@181 | 116 | |
| Recipe-to-image retrieval | Recipe1M 1k setup (test) | Recall@180.3 | 110 |