15 dépôts
Processes for refining neural network parameters using ground-truth data sequences to improve output quality.
Distinct from Model Weight Management: Candidates refer to weights as files or inspections; this refers to the active process of training/optimizing weights.
Explore 15 awesome GitHub repositories matching artificial intelligence & ml · Supervised Model Weight Optimization. Refine with filters or upvote what's useful.
DAIN is a video frame synthesis engine and AI video upsampling tool designed to increase video playback smoothness. It functions as a computer vision model that synthesizes intermediate frames between existing images to transform low frame rate video into high frame rate content. The system utilizes depth-aware video frame interpolation to predict the motion of pixels between consecutive images. By analyzing spatial depth via depth maps, the tool generates new frames that account for occlusions and overlapping objects to create slow motion effects. The framework incorporates optical flow int
Optimizes model weights using triplet datasets and depth networks to improve frame synthesis quality.
This project is a comprehensive collection of educational examples and reference implementations for building vision and language models using PyTorch. It serves as a deep learning tutorial covering the end-to-end process of developing neural networks, from initial architecture definition to final production deployment. The repository provides detailed guides on implementing a wide range of domain-specific models, including convolutional neural networks for object detection and segmentation, as well as transformer and recurrent architectures for natural language processing. It emphasizes gene
Implements supervised weight optimization for translation models using label smoothing and learning rate warmup.
InternLM is a large language model and a comprehensive suite of weights designed for text generation and complex reasoning. It functions as an inference engine for serving responses, a fine-tuning framework for adjusting model weights, and a platform for building autonomous AI agents. The system is capable of processing long-context input sequences up to one million tokens for document analysis. It employs chain-of-thought reasoning to solve knowledge-intensive tasks by generating intermediate logic steps before producing a final answer. The project covers model weight optimization through s
Refines neural network parameters using supervised fine-tuning and RLHF to optimize model behavior.
This project is an object detection framework implementing the YOLOv3 architecture using Keras and TensorFlow. It functions as a deep learning vision model and computer vision toolset designed to locate and classify multiple entities within images and video streams using bounding boxes. The system includes a multi-GPU inference engine to distribute computational loads across several graphics processing units. It also provides a pipeline for creating custom object detectors by retraining pre-trained weights on annotated datasets to recognize user-defined object classes. The framework covers m
Optimizes the model through a two-stage retraining process that stabilizes early layers before full-network fine-tuning.
Ce projet est un générateur de poésie par deep learning conçu pour créer des couplets chinois traditionnels. Il utilise une architecture de réseau de neurones séquence-à-séquence pour mapper des séquences de texte d'entrée vers des séquences de sortie correspondantes, fonctionnant comme un modèle de génération de texte et un service web d'inférence. Le système dispose d'un mécanisme de classement de texte neuronal qui évalue les sorties candidates en fonction de la cohérence de la longueur et des motifs de caractères pour assurer l'alignement structurel. Il inclut également un processus de filtrage de contenu qui scanne le texte généré par rapport à des listes de mots interdits pour supprimer le matériel sensible ou inapproprié. La surface de capacités plus large inclut un pipeline de tokenisation de texte pour la préparation des données et une interface RESTful pour servir les prédictions du modèle via JSON. Le projet couvre également le cycle de vie du machine learning, englobant l'optimisation des poids du modèle, l'entraînement et l'évaluation des performances par rapport aux jeux de données de validation.
Refines neural network parameters, dropout, and learning rates using ground-truth sequences to improve poetry quality.
Composer est un framework d'entraînement distribué PyTorch conçu pour mettre à l'échelle des modèles de grande taille sur des clusters GPU multi-nœuds. Il fonctionne comme un entraîneur de grands modèles de langage, un optimiseur de modèle distribué et un gestionnaire de cycle de vie d'entraînement. Le projet se différencie en tant que bibliothèque de régularisation pour le deep learning, fournissant des techniques d'optimisation spécialisées telles que Sharpness Aware Minimization, MixUp et CutMix pour améliorer la généralisation des modèles. Il distingue davantage son flux d'entraînement par l'utilisation du warmup de longueur de séquence, du gel progressif des couches et du checkpointing d'état fragmenté pour la récupération de modèles à grande échelle. Le framework couvre une large surface de capacités, incluant l'orchestration de l'entraînement distribué, la gestion du matériel en précision mixte et le streaming de données cloud-native. Il fournit également des outils étendus de surveillance et d'observabilité pour les diagnostics de mémoire GPU, la détection de divergence d'entraînement et le suivi du débit. Le projet inclut un lanceur en ligne de commande pour automatiser l'exécution de tâches d'entraînement multi-GPU sur plusieurs nœuds.
Normalizes convolutional weights in a model to improve training stability and convergence.
Flashlight est une bibliothèque de machine learning et de tenseurs autonome en C++ utilisée pour construire et entraîner des réseaux de neurones. Elle fonctionne comme un framework complet de réseaux de neurones et un moteur de différenciation automatique, fournissant les outils pour construire des graphes de calcul et calculer les gradients via la rétropropagation. Le projet sert de framework d'entraînement distribué, utilisant des opérations all-reduce pour synchroniser les gradients et les paramètres sur plusieurs nœuds de calcul et appareils. Il se distingue par une intégration profonde de la manipulation de tenseurs haute performance, l'interopérabilité native de la mémoire des appareils et un système pour synchroniser les poids entre les workers distribués afin d'accélérer l'entraînement de modèles à grande échelle. Le framework couvre un large éventail de capacités de deep learning, incluant la composition modulaire de couches pour concevoir des architectures complexes comme des blocs résiduels et des cellules récurrentes. Il fournit des utilitaires étendus de gestion de données pour l'ingestion et le préchargement, ainsi que des systèmes de sérialisation pour persister les états de modèle. De plus, il inclut une suite d'outils de surveillance et d'observabilité pour suivre les métriques d'entraînement et mesurer les erreurs de séquence. La bibliothèque est implémentée en C++.
Provides weight normalization wrappers for linear and convolutional modules to accelerate model convergence.
Neuraltalk is an automated image captioning system that generates natural language descriptions for images. It utilizes a deep learning model that integrates a pretrained convolutional neural network for visual feature extraction with a recurrent neural network decoder to produce text sequences. The project provides a full workflow for training and evaluating captioning models, including weight optimization via backpropagation and gradient descent. It includes tools for measuring caption accuracy by comparing generated text against reference descriptions. The system covers data preprocessing
Refines neural network parameters using ground-truth data sequences to improve the quality of generated captions.
Ce projet est une implémentation PyTorch de l'architecture EfficientDet conçue pour la détection d'objets en temps réel. Il fournit un réseau neuronal et un moteur d'inférence capable d'identifier et de localiser plusieurs objets au sein d'images ou de flux vidéo. L'implémentation inclut des modèles de vision par ordinateur pré-entraînés avec des poids optimisés, permettant une inférence immédiate et un fine-tuning sans avoir besoin d'entraîner à partir de zéro. Le projet couvre le pipeline complet pour l'optimisation des modèles de vision par ordinateur, incluant l'entraînement à la détection d'objets personnalisés et l'optimisation des poids des modèles. Il incorpore des composants structurels tels que la fusion de caractéristiques bidirectionnelle, des architectures neuronales à échelle composée et des propositions de régions basées sur des ancres pour équilibrer la vitesse d'inférence et la précision de détection.
Refines neural network parameters using ground-truth data and loss balancing to improve custom model performance.
Ce projet est un programme éducatif complet et un framework de deep learning conçu pour enseigner le deep learning pratique avec PyTorch via des notebooks et des exemples de code. Il sert de bibliothèque de haut niveau pour construire, entraîner et déployer des réseaux de neurones, agissant comme un orchestrateur d'entraînement de modèles qui coordonne les modèles PyTorch, les optimiseurs et les fonctions de perte. Le projet fournit des boîtes à outils spécialisées pour la vision par ordinateur, le traitement du langage naturel et le prétraitement de données tabulaires. Il se distingue par des contrôles d'entraînement avancés tels que des taux d'apprentissage discriminatifs, un système de callback bidirectionnel pour personnaliser la logique d'entraînement, et une abstraction de haut niveau qui automatise le placement sur périphérique et les boucles d'entraînement. Le framework couvre une large surface de capacités, y compris la construction automatisée de pipelines de données, l'analyse d'architecture de modèles et l'évaluation des performances sur des tâches de classification, de régression et de segmentation. Il inclut également des utilitaires pour l'entraînement distribué sur plusieurs GPU, l'entraînement en précision mixte pour l'optimisation de la mémoire, et un support spécialisé pour les données d'imagerie médicale. Le projet est livré sous forme d'une série de Jupyter Notebooks.
Updates model parameters using standard optimization algorithms like SGD and Adam with optional weight decay.
Chronos-forecasting is a zero-shot time series forecasting framework based on a pretrained large language model. It enables the prediction of future values across diverse datasets without requiring task-specific training or optimization. The system functions as a probabilistic forecasting tool, producing multiple future trajectories and quantile forecasts to quantify uncertainty and potential prediction errors. It incorporates exogenous covariate integration to merge external variables and historical context into the input stream for increased precision. The project includes utilities for sy
Updates model parameters using specific datasets and configurations to improve prediction accuracy.
This project is a PyTorch sentiment analysis tutorial and a deep learning implementation for analyzing text. It provides a natural language processing sequence classification pipeline designed to clean text data and train neural networks to categorize sequences of words. The implementation focuses on adapting pretrained language models for specific text classification tasks using custom datasets. It includes a process for fine-tuning large-scale language models and implementing recurrent networks and transformers for emotional tone detection. The project covers the broader surface of text se
Refines neural network parameters using supervised gradient descent based on sentiment labels.
MODNet est un modèle de segmentation d'image et de matting de portrait par deep learning conçu pour isoler les sujets humains des arrière-plans. Il génère des mattes alpha de haute qualité pour les images et les vidéos en utilisant uniquement une entrée RGB standard, supprimant l'exigence de guides trimap manuels. Le framework est optimisé pour l'inférence en temps réel et fournit des utilitaires pour exporter les poids de modèles pré-entraînés dans des formats spécialisés pour un déploiement sur le matériel cible. Le projet couvre le flux de travail complet pour l'isolation de portrait, y compris l'entraînement de modèles de matting supervisés sur des jeux de données étiquetés, la suppression d'arrière-plan vidéo en temps réel et le déploiement sur matériel périphérique (edge hardware).
Optimizes model weights for accurate edge detection using supervised training on labeled image pairs.
This project is a collection of structured study notes and notebooks serving as an educational resource for deep learning and neural network fundamentals. It provides a technical reference for implementing machine learning theory, covering everything from basic network design to the construction of advanced architectures. The material specifically focuses on the implementation of convolutional neural networks for computer vision and sequence models for natural language processing. It includes detailed guidance on building object detection systems, face recognition, and speech transcription mo
Provides the fundamental implementation of gradient descent for refining neural network weights and biases.
This repository is an educational collection of implementations and research notes focused on deep learning architectures and optimization techniques. It provides modular code examples designed to demonstrate foundational and advanced concepts in machine learning, ranging from basic neural network structures to complex training strategies. The project distinguishes itself by offering practical implementations of specialized research methods, including capsule-based feature aggregation, gradient direction decoupling, and self-normalizing weight regularization. These materials allow for the stu
Implements weight normalization techniques to maintain statistical distributions and prevent vanishing gradients.