IDEA Prune introduces an integrated enlarge-and-prune pipeline for generative language model pretraining. It combines enlarged model pretraining, pruning, and recovery under a single cosine annealing learning rate schedule, complemented by an iterative structured pruning method for gradual parameter removal. This approach mitigates knowledge loss and facilitates smooth compression and enhanced performance, achieving superior performance of pruned models.
All Apple trails