15 Repos
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.
Dieses Projekt ist ein Deep-Learning-Poetry-Generator, der darauf ausgelegt ist, traditionelle chinesische Couplets zu erstellen. Er nutzt eine Sequence-to-Sequence-Architektur neuronaler Netze, um Eingabetextsequenzen auf passende Ausgabesequenzen zu mappen, und fungiert als Textgenerierungsmodell sowie als Inferenz-Webservice. Das System verfügt über einen neuronalen Text-Ranking-Mechanismus, der Kandidaten-Outputs basierend auf Längenkonsistenz und Zeichenmustern evaluiert, um eine strukturelle Ausrichtung sicherzustellen. Es enthält zudem einen Content-Filter-Prozess, der generierten Text gegen Listen verbotener Wörter scannt, um sensible oder unangemessene Inhalte zu entfernen. Die breitere Funktionspalette umfasst eine Text-Tokenisierungs-Pipeline zur Datenvorbereitung und ein RESTful-Interface zur Bereitstellung von Modellvorhersagen via JSON. Das Projekt deckt zudem den Machine-Learning-Lebenszyklus ab, einschließlich Modellgewichtsoptimierung, Training und Performance-Evaluierung gegen Validierungsdatensätze.
Refines neural network parameters, dropout, and learning rates using ground-truth sequences to improve poetry quality.
Composer ist ein Framework für verteiltes Training mit PyTorch, das für die Skalierung großer Modelle über Multi-Node-GPU-Cluster hinweg entwickelt wurde. Es fungiert als Trainer für Large Language Models, als verteilter Modelloptimierer und als Manager für den Trainingslebenszyklus. Das Projekt hebt sich als Bibliothek für Deep-Learning-Regularisierung hervor und bietet spezialisierte Optimierungstechniken wie Sharpness Aware Minimization, MixUp und CutMix, um die Generalisierung von Modellen zu verbessern. Es differenziert seinen Trainingsablauf zudem durch den Einsatz von Sequence Length Warmup, progressivem Layer-Freezing und Sharded-State-Checkpointing für die Wiederherstellung großer Modelle. Das Framework deckt ein breites Spektrum an Funktionen ab, darunter die Orchestrierung von verteiltem Training, das Management von Mixed-Precision-Hardware und Cloud-natives Daten-Streaming. Es bietet zudem umfangreiche Monitoring- und Observability-Tools für die Diagnose von GPU-Speicher, die Erkennung von Trainingsdivergenz und die Verfolgung des Durchsatzes. Das Projekt enthält einen CLI-Launcher zur Automatisierung der Ausführung von Multi-GPU-Trainingsjobs über mehrere Nodes hinweg.
Normalizes convolutional weights in a model to improve training stability and convergence.
Flashlight ist eine eigenständige C++-Bibliothek für maschinelles Lernen und Tensor-Berechnungen, die zum Erstellen und Trainieren neuronaler Netze verwendet wird. Sie fungiert als umfassendes Framework für neuronale Netze und Engine für automatische Differenzierung und bietet Werkzeuge zur Konstruktion von Berechnungsgraphen und zur Berechnung von Gradienten via Backpropagation. Das Projekt dient als Framework für verteiltes Training und nutzt All-Reduce-Operationen zur Synchronisation von Gradienten und Parametern über mehrere Rechenknoten und Geräte hinweg. Es zeichnet sich durch eine tiefe Integration von leistungsstarker Tensor-Manipulation, nativer Interoperabilität mit Gerätespeichern und einem System zur Synchronisation von Gewichten über verteilte Worker aus, um das Training großskaliger Modelle zu beschleunigen. Das Framework deckt eine breite Palette an Deep-Learning-Funktionen ab, einschließlich modularer Schichtkomposition für den Entwurf komplexer Architekturen wie Residual-Blöcke und rekurrente Zellen. Es bietet umfangreiche Datenmanagement-Utilities für Ingestion und Prefetching sowie Serialisierungssysteme zur Persistierung von Modellzuständen. Zusätzlich enthält es eine Suite an Überwachungs- und Observability-Tools zur Verfolgung von Trainingsmetriken und zur Messung von Sequenzfehlern. Die Bibliothek ist in C++ implementiert.
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.
Dieses Projekt ist eine PyTorch-Implementierung der EfficientDet-Architektur für Echtzeit-Objekterkennung. Es bietet ein neuronales Netzwerk und eine Inference-Engine, die in der Lage ist, mehrere Objekte innerhalb von Bildern oder Videoströmen zu identifizieren und zu lokalisieren. Die Implementierung enthält vortrainierte Computer-Vision-Modelle mit optimierten Gewichten, was eine sofortige Inference und Feinabstimmung ermöglicht, ohne dass ein Training von Grund auf erforderlich ist. Das Projekt deckt die vollständige Pipeline für die Optimierung von Computer-Vision-Modellen ab, einschließlich benutzerdefiniertem Objekterkennungs-Training und Modellgewichts-Optimierung. Es integriert strukturelle Komponenten wie bidirektionales Feature-Fusion, zusammengesetzte neuronale Architekturen und Anchor-basierte Region-Proposals, um Inference-Geschwindigkeit und Erkennungsgenauigkeit auszubalancieren.
Refines neural network parameters using ground-truth data and loss balancing to improve custom model performance.
Dieses Projekt ist ein umfassendes Bildungsprogramm und Deep-Learning-Framework, das darauf ausgelegt ist, praktisches Deep Learning mit PyTorch anhand von Notebooks und Codebeispielen zu vermitteln. Es dient als High-Level-Bibliothek zum Erstellen, Trainieren und Bereitstellen neuronaler Netze und fungiert als Modell-Trainings-Orchestrator, der PyTorch-Modelle, Optimierer und Verlustfunktionen koordiniert. Das Projekt bietet spezialisierte Toolkits für Computer Vision, Natural Language Processing und die Vorverarbeitung tabellarischer Daten. Es zeichnet sich durch fortschrittliche Trainingskontrollen aus, wie z. B. diskriminative Lernraten, ein Zwei-Wege-Callback-System zur Anpassung der Trainingslogik und eine High-Level-Learner-Abstraktion, die die Geräteplatzierung und Trainingsschleifen automatisiert. Das Framework deckt ein breites Fähigkeitsspektrum ab, einschließlich der automatisierten Konstruktion von Datenpipelines, der Analyse von Modellarchitekturen und der Leistungsbewertung bei Klassifizierungs-, Regressions- und Segmentierungsaufgaben. Es enthält zudem Dienstprogramme für verteiltes Training über mehrere GPUs, Mixed-Precision-Training zur Speicheroptimierung und spezialisierte Unterstützung für medizinische Bilddaten. Das Projekt wird als eine Reihe von Jupyter Notebooks bereitgestellt.
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.
Dieses Projekt ist ein PyTorch-Tutorial für Sentiment-Analyse und eine Deep-Learning-Implementierung zur Textanalyse. Es bietet eine NLP-Pipeline (Natural Language Processing) zur Sequenzklassifizierung, die darauf ausgelegt ist, Textdaten zu bereinigen und neuronale Netze für die Kategorisierung von Wortfolgen zu trainieren. Der Fokus liegt auf der Anpassung vortrainierter Sprachmodelle für spezifische Textklassifizierungsaufgaben unter Verwendung benutzerdefinierter Datensätze. Es beinhaltet einen Prozess zum Fine-Tuning großer Sprachmodelle sowie die Implementierung rekurrenter Netze und Transformer zur Erkennung emotionaler Tonalität. Das Projekt deckt ein breites Spektrum der Textsequenzklassifizierung und PyTorch-Textverarbeitung ab. Dies beinhaltet Workflows zur Vorbereitung von Rohtext-Datensätzen mit der TorchText-Bibliothek sowie den Aufbau von Deep-Learning-Modellen zur Kategorisierung von Texten.
Refines neural network parameters using supervised gradient descent based on sentiment labels.
MODNet is a deep learning image segmenter and portrait matting model designed to isolate human subjects from backgrounds. It generates high-quality alpha mattes for images and video using only standard RGB input, removing the requirement for manual trimap guides. The framework is optimized for real-time inference and provides utilities to export pre-trained model weights into specialized formats for deployment on target hardware. The project covers the full workflow for portrait isolation, including supervised matting model training on labeled datasets, real-time video background removal, an
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.