Tinh chỉnh các mô hình học máy như Ultralytics YOLO cho các nhiệm vụ cụ thể. Tìm hiểu các phương pháp, ứng dụng và thực hành tốt nhất tại đây!
Fine-tuning is a popular technique in machine learning (ML) that involves taking a model already trained on a large dataset (a pre-trained model) and further training it on a smaller, specific dataset relevant to a particular task. This approach leverages the general knowledge learned by the model during its initial training, adapting it to excel in a more specialized domain without needing to train a model from scratch. This saves significant time and computational resources, making it a common practice in fields like computer vision (CV) and natural language processing (NLP). Frameworks like PyTorch and TensorFlow provide the tools necessary for implementing fine-tuning.
The process typically starts with selecting a pre-trained model, such as an Ultralytics YOLO model trained on a broad dataset like COCO or ImageNet. These models, often Convolutional Neural Networks (CNNs) for vision or Transformers for NLP, have already learned to recognize general features from their initial training data. During fine-tuning, the model weights—parameters learned during training—are adjusted based on the new, smaller dataset. Often, the initial layers of the network (which learn general features like edges or textures) are kept "frozen" (their weights are not updated), while the later, more task-specific layers are retrained. This retraining usually involves using a lower learning rate than used in the original training to make smaller adjustments to the weights, preserving the previously learned knowledge while adapting to the nuances of the new task. You can find more details on the mechanics in resources like the fast.ai course.
Việc tinh chỉnh mang lại một số lợi thế quan trọng:
Explore additional model training tips for optimizing the process.
Điều quan trọng là phải phân biệt tinh chỉnh với các khái niệm ML tương tự:
Tinh chỉnh được sử dụng rộng rãi trong nhiều lĩnh vực khác nhau:
Explore more applications within the computer vision community.
Ultralytics provides robust support for fine-tuning its YOLO models. Users can easily load pre-trained weights (e.g., from models trained on COCO) and continue training on their own custom dataset for tasks like detection, segmentation, or classification. The Ultralytics documentation offers detailed guides on the training process, enabling users to adapt state-of-the-art models like YOLO11 for their specific computer vision challenges. Platforms like Ultralytics HUB further streamline the custom training process. This adaptability is key to achieving optimal performance in diverse applications, from AI in agriculture to robotics. Further information on transfer learning techniques can be found on educational platforms like Coursera's Deep Learning Specialization and sites like Papers with Code or Distill.pub for research insights.