Optimizing Computational Infrastructure for Large Language Models in Bioinformatics: A Case Study
摘要
This paper addresses the challenge of efficiently training Large Language Models (LLMs) on large-scale, sparse omics datasets in high-performance computing (HPC) environments. Using over 1000 BED tracks as a representative data source, we propose a method combining interval-based chunked storage, sparse matrix transformation, and parallel data loading, integrated within a PyTorch Lightning training framework. Our approach minimizes CPU overhead and disk I/O while preserving data sparsity throughout training. Evaluated on multiple GPU configurations (V100, A100, H100), the method achieves over 85% GPU utilization with less than 20% CPU overhead, demonstrating scalability, adaptability, and potential to significantly accelerate LLM applications in bioinformatics.