nillion aivm

Nillion AIVM

AIVM adalah kerangka kerja mutakhir yang dirancang untuk inferensi pemeliharaan privasi menggunakan protokol kriptografi canggih. Dengan AIVM, Anda dapat menggunakan jaringan pengembangan lokal (DevNet) untuk mengeksplorasi inferensi pribadi menggunakan contoh atau model khusus yang disediakan.

Daftar isi

• Menginstal AIVM
• Menjalankan aivm
• Penggunaan
  • Inferensi pribadi
  • Unggah model khusus
• Lisensi

Menginstal AIVM

1. Buat lingkungan virtual:

   python3 -m venv .venv

2. Aktifkan lingkungan virtual:

   Di Linux/MacOS:

    source .venv/bin/activate

   Di Windows:

   . v env S cripts a ctivate

3. Instal paket:

   Jika Anda akan menjalankan contoh, jalankan:

   pip install " nillion-aivm[examples] "

   Jika tidak, jika Anda akan menghasilkan kode sendiri, Anda bisa hanya:

   pip install nillion-aivm

Menjalankan aivm

1. Mulai AIVM DevNet:

   aivm-devnet

2. Buka contoh notebook jupyter yang disediakan/getting-started.ipynb untuk menjalankan contoh inferensi pribadi di AIVM.

3. Setelah menyelesaikan tugas Anda, hentikan proses DevNet dengan menekan CTRL+C .

Penggunaan

Untuk penggunaan tambahan, lihat folder contoh, yang menunjukkan cara mengatur alur kerja inferensi pribadi menggunakan AIVM.

Melakukan inferensi yang aman

Penggunaan dasar

1. Pertama, impor klien AIVM dan periksa model yang tersedia:

 import aivm_client as aic

# List all supported models
available_models = aic . get_supported_models ()
print ( available_models )

2. Siapkan data input Anda. Berikut contoh menggunakan Pytorch untuk menghasilkan input acak:

 import torch

# Create a sample input (e.g., for LeNet5 MNIST)
random_input = torch . randn (( 1 , 1 , 28 , 28 ))  # Batch size 1, 1 channel, 28x28 pixels

3. Mengenkripsi input Anda menggunakan cryptensor yang sesuai:

 # Encrypt the input
encrypted_input = aic . LeNet5Cryptensor ( random_input )

4. Lakukan inferensi yang aman:

 # Get prediction while maintaining privacy
result = aic . get_prediction ( encrypted_input , "LeNet5MNIST" )

Fungsi get_prediction secara otomatis menangani protokol komputasi yang aman dengan node aivm-devnet , memastikan bahwa data input Anda tetap pribadi di seluruh proses inferensi.

Unggah model khusus

Anda dapat menggunakan model terlatih Anda sendiri ke AIVM, asalkan mereka mengikuti arsitektur yang didukung (Berttiny atau Lenet5).

Mengunggah model khusus

1. Impor klien AIVM:

 import aivm_client as aic

2. Unggah model kustom Anda:

 # For BertTiny models
aic . upload_bert_tiny_model ( model_path , "MyCustomBertTiny" )

# For LeNet5 models
aic . upload_lenet5_model ( model_path , "MyCustomLeNet5" )

3. Lakukan inferensi dengan model khusus Anda:

 # For BertTiny models
result = aic . get_prediction ( private_berttiny_input , "MyCustomBertTiny" )

# For LeNet5 models
result = aic . get_prediction ( private_lenet5_input , "MyCustomLeNet5" )

Lisensi

Proyek ini dilisensikan di bawah lisensi MIT.