คู่มือทางเทคนิค

ค่าเฉลี่ยน้ำหนักสุ่ม

Stochastic Weight Averaging (SWA) ใช้ค่าเฉลี่ยอย่างง่ายของน้ำหนักของแบบจำลองจากหลายจุดที่ล่าช้าในการฝึก แทนที่จะเก็บแค่สแน็ปช็อตสุดท้าย

ภาพรวม

Stochastic Weight Averaging (SWA) ใช้ค่าเฉลี่ยอย่างง่ายของน้ำหนักของแบบจำลองจากหลายจุดที่ล่าช้าในการฝึก แทนที่จะเก็บแค่สแน็ปช็อตสุดท้าย เคล็ดลับราคาถูกนี้มักจะทำให้โมเดลอยู่ในขอบเขตที่ราบเรียบและกว้างกว่าของแนวการสูญเสีย ซึ่งมีแนวโน้มที่จะสรุปข้อมูลทั่วไปได้ดีขึ้นอย่างเห็นได้ชัดในข้อมูลที่มองไม่เห็น

Stochastic Weight Averaging เป็นองค์ประกอบทางเทคนิคที่ส่งผลต่อคุณภาพของโมเดล ต้นทุนโครงสร้างพื้นฐาน เวลาแฝง และความน่าเชื่อถือในระดับต่างๆ

เจาะลึก

SWA เปิดตัวโดย Izmailov, Wilson และเพื่อนร่วมงานในปี 2018 โดยใช้ประโยชน์จากข้อสังเกตที่ว่า SGD ที่มีอัตราการเรียนรู้คงที่หรือเป็นวัฏจักรไม่ได้มาบรรจบกันที่จุดเดียว แต่จะเด้งไปรอบขอบหุบเขาที่ราบเรียบกว้าง แทนที่จะเลือกจุดหยุดที่มีเสียงดังจุดใดจุดหนึ่ง SWA จะใช้อัตราการเรียนรู้ที่สูงปานกลาง (มักจะคงที่หรือเป็นรอบ) สำหรับยุคสุดท้าย และเฉลี่ยน้ำหนักที่เข้าชม ซึ่งโดยทั่วไปจะเป็นทุกยุค ตุ้มน้ำหนักเฉลี่ยจะอยู่ใกล้กับศูนย์กลางของพื้นที่ราบมากกว่า เนื่องจากสถิติการทำให้เป็นมาตรฐานแบบแบตช์ได้รับการคำนวณสำหรับน้ำหนักเฉพาะ SWA จึงต้องมีการส่งต่อข้อมูลเพิ่มเติมหนึ่งครั้งเพื่อคำนวณวิธีการทำงานและความแปรปรวนของ BN ใหม่สำหรับแบบจำลองโดยเฉลี่ย โดยพื้นฐานแล้วไม่มีค่าใช้จ่าย และความแม่นยำที่เพิ่มขึ้นจะสม่ำเสมอทั่วทั้งตัวแยกประเภทรูปภาพและนอกเหนือจากนั้น

ข้อมูลเชิงลึกทางเทคนิค

SWA รักษาค่าเฉลี่ยรันอยู่ w_SWA = (n·w_SWA + w_i)/(n+1) อัปเดตในแต่ละรอบ ในขณะที่โมเดล SGD แบบสดยังคงสำรวจต่อไปด้วยอัตราการเรียนรู้ที่ค่อนข้างสูง ค่าเฉลี่ยในพื้นที่น้ำหนักจะประมาณค่าทั้งมวลในพื้นที่ฟังก์ชัน แต่ใช้โมเดลหนึ่งในการอนุมาน ไม่ใช่จำนวนมาก กลไกสำคัญคือค่าขั้นต่ำแบบเรียบนั้นทนทานต่อการเปลี่ยนแปลงของน้ำหนัก ดังนั้นพื้นผิวการสูญเสียการฝึกอบรม/การทดสอบจึงอยู่ในแนวเดียวกัน ช่วยลดช่องว่างทั่วไป

การเรียนรู้ค่าเฉลี่ยน้ำหนักสุ่ม

Stochastic Weight Averaging (SWA) ใช้ค่าเฉลี่ยอย่างง่ายของน้ำหนักของแบบจำลองจากหลายจุดที่ล่าช้าในการฝึก แทนที่จะเก็บแค่สแน็ปช็อตสุดท้าย เคล็ดลับราคาถูกนี้มักจะทำให้โมเดลอยู่ในขอบเขตที่ราบเรียบและกว้างกว่าของแนวการสูญเสีย ซึ่งมีแนวโน้มที่จะสรุปข้อมูลทั่วไปได้ดีขึ้นอย่างเห็นได้ชัดในข้อมูลที่มองไม่เห็น Stochastic Weight Averaging เป็นองค์ประกอบทางเทคนิคที่ส่งผลต่อคุณภาพของโมเดล ต้นทุนโครงสร้างพื้นฐาน เวลาแฝง และความน่าเชื่อถือในระดับต่างๆ เพื่อสร้างความเข้าใจอย่างลึกซึ้ง ให้ถือว่า Stochastic Weight Averaging เป็นรูปแบบการดำเนินงาน ไม่ใช่คุณลักษณะเดียว: กำหนดผลลัพธ์ที่ต้องการ ชี้แจงสมมติฐาน และแยกสิ่งที่ระบบสามารถทำได้อย่างน่าเชื่อถือจากสิ่งที่ยังต้องใช้วิจารณญาณจากผู้เชี่ยวชาญ

ในทางปฏิบัติ ทีมที่แข็งแกร่งที่ใช้ Stochastic Weight Averaging จะปรับสถาปัตยกรรม ข้อมูล และตัวเลือกโครงสร้างพื้นฐานให้เหมาะสมโดยเทียบกับความน่าเชื่อถือและต้นทุน โดยจะบันทึกเกณฑ์ความสำเร็จที่ชัดเจน ทดสอบกับข้อมูลและขั้นตอนการทำงานที่สมจริง และทำซ้ำตามรูปแบบความล้มเหลวที่สังเกตได้ แทนที่จะชนะการวัดประสิทธิภาพเพียงครั้งเดียว นี่คือจุดที่ความเข้าใจทางทฤษฎีกลายเป็นความสามารถที่คงทนของผลิตภัณฑ์ นโยบาย และการดำเนินงาน

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี ในเวลาเดียวกัน การเพิ่มประสิทธิภาพเกณฑ์มาตรฐานหนึ่งรายการสามารถซ่อนจุดอ่อนของระบบในวงกว้างได้ แนวทางที่ยืดหยุ่นที่สุดคือการรวมความเร็วของการทดลองเข้ากับวินัยในการกำกับดูแล: ดำเนินการนำร่อง จับหลักฐาน เผยแพร่บันทึกการตัดสินใจ และอัปเดตการป้องกันอย่างต่อเนื่องเมื่อพฤติกรรมของโมเดล ความคาดหวังของผู้ใช้ และข้อกำหนดด้านกฎระเบียบมีการเปลี่ยนแปลง

ผลกระทบเชิงกลยุทธ์

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

การศึกษาด้านเทคนิคช่วยให้ทีมเลือกกลุ่มที่เหมาะสม ไม่ใช่แค่กลุ่มใหม่ล่าสุด

การศึกษาด้านเทคนิคช่วยให้ทีมเลือกกลุ่มที่เหมาะสม ไม่ใช่แค่กลุ่มใหม่ล่าสุด ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

ตัวเลือกทางวิศวกรรมที่ดีกว่าจะช่วยลดเหตุการณ์ด้านความน่าเชื่อถือในการผลิต

ตัวเลือกทางวิศวกรรมที่ดีกว่าจะช่วยลดเหตุการณ์ด้านความน่าเชื่อถือในการผลิต ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

อนาคตของการเฉลี่ยน้ำหนักสุ่ม

SWA ได้สร้างตัวแปรต่างๆ เช่น SWA-Gaussian (SWAG) สำหรับความไม่แน่นอนแบบเบย์เซียนราคาถูก และแนวคิดในการหาค่าเฉลี่ยในขณะนี้เป็นรากฐานของเทคนิค Exponential Moving Average ที่ใช้กันอย่างแพร่หลายในแบบจำลองการแพร่กระจาย การเรียนรู้ด้วยตนเอง และการฝึกล่วงหน้าด้วยแบบจำลองขนาดใหญ่ คาดว่าการเฉลี่ยน้ำหนักจะยังคงเป็น 'อาหารกลางวันฟรี' เริ่มต้นในสูตรอาหารการฝึกอบรม โดยมีการวิจัยขยายไปสู่การผสานแบบจำลองที่ได้รับการฝึกอบรมอย่างอิสระ (ซุปแบบจำลอง) และปรับปรุงการสอบเทียบควบคู่ไปกับความแม่นยำของวัตถุดิบ

การใช้งานจริงในโลกแห่งความเป็นจริง

เพิ่มความแม่นยำในการทดสอบของตัวแยกประเภทรูปภาพ ResNet และ DenseNet บน CIFAR และ ImageNet โดยไม่มีค่าใช้จ่ายเพิ่มเติมในการอนุมาน

SWAG (SWA-Gaussian) สร้างค่าประมาณความไม่แน่นอนที่ปรับเทียบแล้วสำหรับการคาดการณ์ที่คำนึงถึงความปลอดภัยจากการฝึกซ้อมเพียงครั้งเดียว

EMA-of-weights ที่ทำให้เครือข่ายการสุ่มตัวอย่างมีเสถียรภาพในเครื่องกำเนิดภาพแบบแพร่ เช่น Stable Diffusion

การสร้าง 'ซุปจำลอง' โดยเฉลี่ยจุดตรวจสอบที่ได้รับการปรับแต่งอย่างละเอียดหลายแห่งเพื่อปรับปรุงความแข็งแกร่งโดยไม่ต้องฝึกอบรมใหม่

รูปแบบการดำเนินงาน

ค่าเฉลี่ยน้ำหนักสุ่มในทางปฏิบัติ

เพิ่มความแม่นยำในการทดสอบของตัวแยกประเภทรูปภาพ ResNet และ DenseNet บน CIFAR และ ImageNet โดยไม่มีค่าใช้จ่ายเพิ่มเติมในการอนุมาน

การเพิ่มความแม่นยำในการทดสอบของตัวแยกประเภทรูปภาพ ResNet และ DenseNet บน CIFAR และ ImageNet โดยไม่มีค่าใช้จ่ายการอนุมานเพิ่มเติม ทีมมักจะได้รับผลลัพธ์ที่ดีกว่าเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการทำงานที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

ค่าเฉลี่ยน้ำหนักสุ่มในทางปฏิบัติ

SWAG (SWA-Gaussian) สร้างค่าประมาณความไม่แน่นอนที่ปรับเทียบแล้วสำหรับการคาดการณ์ที่คำนึงถึงความปลอดภัยจากการฝึกซ้อมเพียงครั้งเดียว

SWAG (SWA-Gaussian) ที่สร้างค่าประมาณความไม่แน่นอนที่ปรับเทียบแล้วสำหรับการคาดการณ์ที่คำนึงถึงความปลอดภัยจากการฝึกอบรมเพียงครั้งเดียว ทีมมักจะได้รับผลลัพธ์ที่ดีกว่าเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการผลิตที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

ค่าเฉลี่ยน้ำหนักสุ่มในทางปฏิบัติ

EMA-of-weights ที่ทำให้เครือข่ายการสุ่มตัวอย่างมีเสถียรภาพในเครื่องกำเนิดภาพแบบแพร่ เช่น Stable Diffusion

EMA-of-weights ที่ทำให้เครือข่ายการสุ่มตัวอย่างมีเสถียรภาพในโปรแกรมสร้างภาพแบบกระจาย เช่น Stable Diffusion Teams มักจะได้รับผลลัพธ์ที่ดีกว่า เมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการทำงานที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

ค่าเฉลี่ยน้ำหนักสุ่มในทางปฏิบัติ

การสร้าง 'ซุปจำลอง' โดยเฉลี่ยจุดตรวจสอบที่ได้รับการปรับแต่งอย่างละเอียดหลายแห่งเพื่อปรับปรุงความแข็งแกร่งโดยไม่ต้องฝึกอบรมใหม่

การสร้าง 'ซุปแบบจำลอง' โดยเฉลี่ยจุดตรวจที่ได้รับการปรับแต่งหลายจุดเพื่อปรับปรุงความแข็งแกร่งโดยไม่ต้องฝึกอบรมใหม่ ทีมมักจะได้รับผลลัพธ์ที่ดีกว่าเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งการเพิ่มผลผลิตและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

ความเสี่ยงและรั้ว

!

การเพิ่มประสิทธิภาพเกณฑ์มาตรฐานหนึ่งรายการสามารถซ่อนจุดอ่อนของระบบในวงกว้างได้

!

ต้นทุนโครงสร้างพื้นฐานและการบำรุงรักษามักถูกประเมินต่ำไป

!

ช่องว่างด้านความปลอดภัยและความสามารถในการสังเกตสามารถเพิ่มขึ้นได้เมื่อระบบมีความซับซ้อนมากขึ้น

แผนงานการดำเนินงาน

1

กำหนดเป้าหมายเวลาแฝง คุณภาพ และต้นทุนก่อนนำไปใช้งาน

กำหนดเป้าหมายเวลาแฝง คุณภาพ และต้นทุนก่อนนำไปใช้งาน ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

2

เกณฑ์มาตรฐานภายใต้สภาวะโหลดและข้อมูลจริง

เกณฑ์มาตรฐานภายใต้สภาวะโหลดและข้อมูลจริง ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

3

การตรวจสอบเครื่องมือเพื่อหาข้อผิดพลาด การเบี่ยงเบน และผลกระทบต่อผู้ใช้

การตรวจสอบเครื่องมือเพื่อหาข้อผิดพลาด การเบี่ยงเบน และผลกระทบต่อผู้ใช้ ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

4

เตรียมเส้นทางการย้อนกลับและการตอบสนองต่อเหตุการณ์ก่อนปรับขนาด

เตรียมเส้นทางการย้อนกลับและการตอบสนองต่อเหตุการณ์ก่อนปรับขนาด ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

สำรวจต่อไป