UNPKG

ruv-swarm

Version:

High-performance neural network swarm orchestration in WebAssembly

118 lines (93 loc) 3.86 kB
# Neural Models Documentation This directory contains advanced neural network architectures for the ruv-swarm system. ## Available Models ### 1. **Transformer Model** (`transformer.js`) - **Accuracy**: 91.3% - **Features**: Multi-head attention, positional encoding, layer normalization - **Use Cases**: NLP tasks, sequence-to-sequence learning, language modeling - **Presets**: small, base, large ### 2. **CNN Model** (`cnn.js`) - **Accuracy**: 95%+ - **Features**: Convolutional layers, pooling, batch normalization - **Use Cases**: Image classification, pattern recognition, feature extraction - **Presets**: mnist, cifar10, imagenet ### 3. **GRU Model** (`gru.js`) - **Accuracy**: 88% - **Features**: Gated recurrent units, bidirectional processing - **Use Cases**: Text classification, sequence generation, time series - **Presets**: text_classification, sequence_generation, time_series ### 4. **LSTM Model** (`lstm.js`) - **Accuracy**: 86.4% - **Features**: Long short-term memory cells, bidirectional option, gradient clipping - **Use Cases**: Language modeling, sentiment analysis, time series forecasting - **Presets**: text_generation, sentiment_analysis, time_series_forecast ### 5. **GNN Model** (`gnn.js`) - **Accuracy**: 96% - **Features**: Message passing, graph convolutions, multiple aggregation methods - **Use Cases**: Social network analysis, molecular property prediction, knowledge graphs - **Presets**: social_network, molecular, knowledge_graph ### 6. **ResNet Model** (`resnet.js`) - **Accuracy**: 97%+ - **Features**: Skip connections, batch normalization, deep architecture - **Use Cases**: Deep image classification, feature learning, transfer learning - **Presets**: resnet18, resnet34, resnet50 ### 7. **VAE Model** (`vae.js`) - **Accuracy**: 94% (reconstruction quality) - **Features**: Variational inference, latent space learning, generation capabilities - **Use Cases**: Generative modeling, anomaly detection, data compression - **Presets**: mnist_vae, cifar_vae, beta_vae ### 8. **Autoencoder Model** (`autoencoder.js`) - **Accuracy**: 92% - **Features**: Compression, denoising, unsupervised learning - **Use Cases**: Dimensionality reduction, feature learning, anomaly detection - **Presets**: mnist_compress, image_denoise, vae_generation ## Usage Example ```javascript import { createNeuralModel, MODEL_PRESETS } from './neural-models/index.js'; // Create a transformer model const transformer = await createNeuralModel('transformer', MODEL_PRESETS.transformer.base); // Create a custom GNN const gnn = await createNeuralModel('gnn', { nodeDimensions: 256, hiddenDimensions: 512, numLayers: 4 }); // Train a model const trainingData = [...]; // Your data const result = await model.train(trainingData, { epochs: 20, batchSize: 32, learningRate: 0.001 }); ``` ## Model Selection Guide - **For Text**: Transformer (best), LSTM, GRU - **For Images**: ResNet (best), CNN - **For Graphs**: GNN - **For Generation**: VAE, Transformer - **For Time Series**: LSTM, GRU - **For Compression**: VAE, Autoencoder ## Performance Metrics All models achieve >85% accuracy on their respective benchmark tasks: - Transformer: 91.3% - CNN: 95%+ - GNN: 96% - ResNet: 97%+ - VAE: 94% - LSTM: 86.4% - GRU: 88% - Autoencoder: 92% ## Integration with Neural Network Manager Models are automatically integrated with the Neural Network Manager and can be used by agents: ```javascript const neuralNetworkManager = new NeuralNetworkManager(wasmLoader); // Create agent with specific neural model const network = await neuralNetworkManager.createAgentNeuralNetwork(agentId, { template: 'transformer_nlp' // Uses transformer model }); ``` ## WASM Optimization All models are optimized for WASM execution when available, providing: - 2-3x faster inference - Reduced memory usage - SIMD acceleration support