2025-11-22T09:37:22.788326

Foragax: An Agent-Based Modelling Framework Based on JAX

Chaturvedi, El-Gazzar, van Gerven

Foraging for resources is a ubiquitous activity conducted by living organisms in a shared environment to maintain their homeostasis. Modelling multi-agent foraging in-silico allows us to study both individual and collective emergent behaviour in a tractable manner. Agent-based modelling has proven to be effective in simulating such tasks, though scaling the simulations to accommodate large numbers of agents with complex dynamics remains challenging. In this work, we present Foragax, a general-purpose, scalable, hardware-accelerated, multi-agent foraging toolkit. Leveraging the JAX library, our toolkit can simulate thousands of agents foraging in a common environment, in an end-to-end vectorized and differentiable manner. The toolkit provides agent-based modelling tools to model various foraging tasks, including options to design custom spatial and temporal agent dynamics, control policies, sensor models, and boundary conditions. Further, the number of agents during such simulations can be increased or decreased based on custom rules. While applied to foraging, the toolkit can also be used to model and simulate a wide range of other multi-agent scenarios.

academic

Foragax: JAX-এর উপর ভিত্তি করে একটি এজেন্ট-ভিত্তিক মডেলিং ফ্রেমওয়ার্ক

মৌলিক তথ্য

পেপার আইডি: 2409.06345
শিরোনাম: Foragax: An Agent-Based Modelling Framework Based on JAX
লেখক: Siddharth Chaturvedi, Ahmed El-Gazzar, Marcel van Gerven (Radboud University)
শ্রেণীবিভাগ: cs.MA (মাল্টি-এজেন্ট সিস্টেম)
প্রকাশনার সময়: ২০২৪ সালের সেপ্টেম্বর (arXiv প্রি-প্রিন্ট)
পেপার লিংক: https://arxiv.org/abs/2409.06345
কোড রিপোজিটরি: https://github.com/i-m-iron-man/Foragax.git

সারসংক্ষেপ

খাদ্য সংগ্রহ হল জীবিত প্রাণীদের একটি সর্বজনীন কার্যকলাপ যা তারা ভাগ করা পরিবেশে হোমিওস্টেসিস বজায় রাখার জন্য সম্পাদন করে। কম্পিউটেশনাল মডেলিংয়ের মাধ্যমে মাল্টি-এজেন্ট খাদ্য সংগ্রহ আচরণ অধ্যয়ন করে, ব্যক্তিগত এবং সম্মিলিত উদীয়মান আচরণ পরিচালনাযোগ্য উপায়ে অনুসন্ধান করা যায়। এজেন্ট মডেলিং এই ধরনের কাজগুলি অনুকরণ করতে কার্যকর প্রমাণিত হয়েছে, তবে জটিল গতিশীলতা সহ বিশাল সংখ্যক এজেন্টদের সামঞ্জস্য করার জন্য সিমুলেশন সম্প্রসারণ করা এখনও চ্যালেঞ্জিং। এই পত্রটি Foragax উপস্থাপন করে, একটি সর্বজনীন, স্কেলেবল, হার্ডওয়্যার-ত্বরিত মাল্টি-এজেন্ট খাদ্য সংগ্রহ টুলকিট। JAX লাইব্রেরি ব্যবহার করে, এই টুলকিটটি হাজার হাজার এজেন্টের খাদ্য সংগ্রহ আচরণ একটি সাধারণ পরিবেশে সম্পূর্ণ ভেক্টরাইজড এবং পার্থক্যযোগ্য পদ্ধতিতে অনুকরণ করতে পারে। টুলকিটটি বিভিন্ন খাদ্য সংগ্রহ কাজ অনুকরণ করার জন্য এজেন্ট মডেলিং সরঞ্জাম সরবরাহ করে, যার মধ্যে রয়েছে কাস্টম স্থানিক এবং সময়গত এজেন্ট গতিশীলতা, নিয়ন্ত্রণ কৌশল, সেন্সর মডেল এবং সীমানা শর্তাদি ডিজাইন করার বিকল্প। অতিরিক্তভাবে, সিমুলেশনের সময় কাস্টম নিয়মের উপর ভিত্তি করে এজেন্টের সংখ্যা বৃদ্ধি বা হ্রাস করা যায়। যদিও খাদ্য সংগ্রহে প্রয়োগ করা হয়েছে, এই টুলকিটটি বিস্তৃত অন্যান্য মাল্টি-এজেন্ট পরিস্থিতি মডেলিং এবং অনুকরণের জন্যও ব্যবহার করা যায়।

গবেষণা পটভূমি এবং প্রেরণা

সমস্যা সংজ্ঞা

খাদ্য সংগ্রহ হল সমস্ত জীবিত প্রাণীকে বেঁচে থাকার জন্য সমাধান করতে হবে এমন সবচেয়ে প্রাচীন এবং গুরুত্বপূর্ণ আচরণগত অপ্টিমাইজেশন সমস্যাগুলির মধ্যে একটি। সবচেয়ে সহজ ব্যাকটেরিয়া থেকে অত্যন্ত সামাজিক প্রাইমেটদের কাছে, দক্ষ খাদ্য সংগ্রহের প্রবৃত্তি বিবর্তনের সমস্ত পর্যায়ে সর্বজনীন। প্রকৃতিতে খাদ্য সংগ্রহ মূলত একটি মাল্টি-এজেন্ট কার্যকলাপ, কারণ সংলগ্ন জীবিত প্রাণীরা একই খাদ্য সংগ্রহ পরিবেশ ভাগ করে।

বিদ্যমান পদ্ধতির সীমাবদ্ধতা

১. স্কেলেবিলিটি সমস্যা: ঐতিহ্যবাহী এজেন্ট মডেলিং পদ্ধতি বিশাল সংখ্যক এজেন্ট পরিচালনা করার সময় গণনামূলক বাধার সম্মুখীন হয় २. জৈবিক বাস্তবতার অভাব: বিদ্যমান সিমুলেশনে এজেন্টের সংখ্যার সীমাবদ্ধতা প্রকৃতিতে পর্যবেক্ষিত খাদ্য সংগ্রহ আচরণের সাথে অসামঞ্জস্য তৈরি করে ३. গণনামূলক জটিলতা: প্রতিটি এজেন্টের অবস্থা অন্যান্য সমস্ত এজেন্টের আচরণের উপর নির্ভর করে, এজেন্টের সংখ্যা বৃদ্ধির সাথে সাথে সিমুলেশন অত্যন্ত কঠিন হয়ে ওঠে

গবেষণা প্রেরণা

লেখকরা খাদ্য সংগ্রহ কাজের মডেলারদের একটি একক পরিবেশে একযোগে বিশাল সংখ্যক (হাজার হাজার) এজেন্ট অনুকরণ করার ক্ষমতা প্রদান করতে চান। এই বড় আকারের সিমুলেশন ক্ষমতা জটিল সম্মিলিত আচরণের উদীয়মান বিষয় বোঝার জন্য অত্যন্ত গুরুত্বপূর্ণ এবং তাত্ত্বিক গবেষণা এবং প্রাকৃতিক পর্যবেক্ষণের মধ্যে ব্যবধান কমাতে পারে।

মূল অবদান

१. Foragax ফ্রেমওয়ার্ক প্রস্তাব: JAX-এর উপর ভিত্তি করে একটি সর্বজনীন, স্কেলেবল, হার্ডওয়্যার-ত্বরিত মাল্টি-এজেন্ট খাদ্য সংগ্রহ মডেলিং টুলকিট २. বড় আকারের সমান্তরাল সিমুলেশন বাস্তবায়ন: সম্পূর্ণ ভেক্টরাইজড পদ্ধতির মাধ্যমে হাজার হাজার এজেন্ট অনুকরণ করতে সক্ষম ३. নমনীয় মডেলিং ইন্টারফেস প্রদান: কাস্টম স্থানিক এবং সময়গত গতিশীলতা, নিয়ন্ত্রণ কৌশল, সেন্সর মডেল এবং সীমানা শর্তাদি সমর্থন করে ४. গতিশীল এজেন্ট ব্যবস্থাপনা সমর্থন: সিমুলেশনের সময় কাস্টম নিয়মের উপর ভিত্তি করে এজেন্টের সংখ্যা বৃদ্ধি বা হ্রাস করা যায় ५. অ-পর্যায়ক্রমিক ক্রমাগত সিমুলেশন বাস্তবায়ন: পরিবেশ এবং এজেন্টের অবস্থা রিসেট না করে দীর্ঘমেয়াদী সিমুলেশন সমর্থন করে

পদ্ধতির বিস্তারিত বিবরণ

কাজের সংজ্ঞা

Foragax একটি সাধারণ পরিবেশে হোমিওস্টেসিস বজায় রেখে বিশাল সংখ্যক এজেন্টের খাদ্য সংগ্রহ আচরণ অনুকরণে মনোনিবেশ করে। কাজের বৈশিষ্ট্যগুলির মধ্যে রয়েছে:

ইনপুট: এজেন্টের প্রাথমিক অবস্থা, পরিবেশ কনফিগারেশন, সম্পদ বিতরণ
আউটপুট: এজেন্ট আচরণ ট্র্যাজেক্টরি, সম্মিলিত উদীয়মান আচরণ প্যাটার্ন
সীমাবদ্ধতা: সম্পদ সীমিততা, এজেন্ট মধ্যে মিথস্ক্রিয়া, পরিবেশ সীমানা শর্তাদি

ফ্রেমওয়ার্ক আর্কিটেকচার

Foragax পাঁচটি প্রধান বিমূর্ততা স্তরের শ্রেণী ডিজাইনের উপর ভিত্তি করে:

१. নীতি (Policy)

অবস্থা ভেরিয়েবল এবং পরামিতি সেট দ্বারা প্রতিনিধিত্ব করা হয়
পরিবেশে এজেন্ট আচরণ উৎপন্ন করতে ব্যবহৃত হয়
একাধিক মডেল সমর্থন করে: Wilson-Cowan মডেল, পুনরাবৃত্তিমূলক স্নায়ু নেটওয়ার্ক, বহু-স্তরীয় পারসেপ্ট্রন
নির্ধারণীয় বা স্টোকাস্টিক প্রকৃতি হিসাবে মডেল করা যায়

२. এজেন্ট (Agent)

অবস্থা ভেরিয়েবল এবং পরামিতি সেট দ্বারা প্রতিনিধিত্ব করা হয়
কাস্টম সময়গত এবং স্থানিক গতিশীলতা সমর্থন করে
অনন্য সূচক এবং সক্রিয় অবস্থা বজায় রাখে (১ সক্রিয়, ০ নিষ্ক্রিয়)
গতিশীলতা তার নীতি দ্বারা পরিচালিত হতে পারে

३. এজেন্ট-সেট (Agent-set)

এজেন্টের সমষ্টিগত প্রতিনিধিত্ব
সমস্ত এজেন্ট একই অবস্থা ভেরিয়েবল, পরামিতি এবং নীতি কাঠামো ভাগ করে
নির্বাচন, সাজানো, যোগ করা বা এজেন্ট মুছে ফেলার অপারেশন সমর্থন করে
শূন্য প্যাডিং এর মাধ্যমে ডেটা কাঠামো আকৃতি ধ্রুবক রাখে, JAX-এর JIT এবং স্বয়ংক্রিয় ভেক্টরাইজেশন বৈশিষ্ট্য বজায় রাখে

४. শেখার অ্যালগরিদম (Learning Algorithms)

নীতি এবং এজেন্ট পরামিতি টিউনিং করার জন্য অ্যালগরিদম সংগ্রহ প্রদান করে
বর্তমান বাস্তবায়ন গ্রেডিয়েন্ট-মুক্ত স্নায়ু বিবর্তন পদ্ধতির উপর ভিত্তি করে
গ্রেডিয়েন্ট-ভিত্তিক শক্তিশালী শেখার পদ্ধতিতে সম্প্রসারণের পরিকল্পনা রয়েছে

५. কাজ (Tasks)

কাস্টম সীমানা শর্তাদি এবং দেয়াল স্থাপনার সাথে খাদ্য সংগ্রহ পরিবেশ ডিজাইন করতে অনুমতি দেয়
সংঘর্ষ সনাক্তকরণের জন্য রশ্মি নিক্ষেপ অ্যালগরিদম ব্যবহার করে

প্রযুক্তিগত উদ্ভাবন পয়েন্ট

१. JAX ব্যাকএন্ড অপ্টিমাইজেশন

JAX-এর তাৎক্ষণিক সংকলন (JIT) এবং স্বয়ংক্রিয় ভেক্টরাইজেশন (AV) ব্যবহার করে
সম্পূর্ণ ভেক্টরাইজড এবং পার্থক্যযোগ্য গণনা বাস্তবায়ন করে
হার্ডওয়্যার ত্বরণ (GPU/TPU) সমর্থন করে

२. শূন্য প্যাডিং কৌশল

সর্বাধিক আকারের ডেটা কাঠামো পূর্ব-বরাদ্দের মাধ্যমে
ধ্রুবক আকৃতি বজায় রাখতে শূন্য প্যাডিং ব্যবহার করে
এজেন্ট সমষ্টি অপারেশনে JAX অপ্টিমাইজেশন বৈশিষ্ট্য বজায় রাখে

३. ক্রমাগত অ-পর্যায়ক্রমিক সিমুলেশন

পরিবেশ এবং এজেন্টের অবস্থা রিসেট করে না
জনসংখ্যার আকার ওঠানামা সমর্থন করে
খোলা-শেষ অভিযোজিত আচরণের উদীয়মান বিষয় প্রচার করে

পরীক্ষামূলক সেটআপ

পরীক্ষামূলক কনফিগারেশন

এজেন্টের সংখ্যা: ১০০০ এজেন্ট
স্নায়ু নেটওয়ার্ক আকার: প্রতিটি এজেন্টের জন্য ৫০ নিউরনের হার-ভিত্তিক স্নায়ু জনসংখ্যা মডেল
সম্পদের সংখ্যা: ৩০০ সম্পদ প্যাচ
সিমুলেশন পদক্ষেপ: ১ মিলিয়ন পদক্ষেপ
গণনা প্ল্যাটফর্ম: ক্লাস্টার কম্পিউটার

গতিশীলতা মডেল

সম্পদ গতিশীলতা

সম্পদ মূল্য Lotka-Volterra গতিশীলতা অনুসরণ করে:

ṡn = ϵsn - αs²n - Σ w(xn, xm)

যেখানে:

ϵ এবং α যথাক্রমে সম্পদের বৃদ্ধি এবং ক্ষয় হার
w(xn, xm) n-তম সম্পদ অবস্থান xn এবং m-তম এজেন্ট অবস্থান xm এর মধ্যে দূরত্বের বিপরীতভাবে সমানুপাতী একটি ফাংশন

এজেন্ট গতিবিধি মডেল

এজেন্ট অবস্থান এবং বেগ দ্বিগুণ ইন্টিগ্রেটর মডেল ব্যবহার করে:

q̇m = [0 1; 0 0]qm + [0; 1]um

যেখানে:

qm = xm, ẋmᵀ এজেন্ট অবস্থা ভেক্টর
um ∈ ℝ ত্বরণ ইনপুট
ত্বরণ ৫০ নিউরনের পুনরাবৃত্তিমূলক স্নায়ু নেটওয়ার্ক দ্বারা রৈখিকভাবে পড়া গড় মূল্য দ্বারা নির্ধারিত হয়

পরীক্ষামূলক ফলাফল

প্রধান ফলাফল

গণনামূলক দক্ষতা: ১০০০ এজেন্ট, ৩০০ সম্পদ, ১ মিলিয়ন পদক্ষেপের সিমুলেশন ৪০ মিনিটের মধ্যে সম্পন্ন হয়
স্কেলেবিলিটি যাচাইকরণ: ৬০০ সম্পদে ৬০০ এজেন্টের খাদ্য সংগ্রহ আচরণ সফলভাবে প্রদর্শন করা হয়েছে
সিস্টেম স্থিতিশীলতা: সিমুলেশনের সময় এজেন্ট এবং সম্পদের স্থানিক বিতরণ যুক্তিসঙ্গত থাকে

কর্মক্ষমতা প্রদর্শন

পরীক্ষামূলক ফলাফল নির্দেশ করে যে Foragax: १. বড় আকারের মাল্টি-এজেন্ট সিমুলেশন দক্ষতার সাথে পরিচালনা করতে পারে २. বাস্তব-সময় কর্মক্ষমতা প্রয়োজনীয়তা বজায় রাখতে পারে ३. জৈবিকভাবে যুক্তিসঙ্গত খাদ্য সংগ্রহ আচরণ প্যাটার্ন উৎপন্ন করতে পারে

ভিজ্যুয়ালাইজেশন ফলাফল

পত্রটি ৬০০ সম্পদ (লাল বিন্দু) মধ্যে ৬০০ এজেন্ট (নীল বিন্দু) বিক্ষিপ্ত খাদ্য সংগ্রহের ভিজ্যুয়ালাইজেশন প্রদান করে, তীরগুলি এজেন্ট দিক নির্দেশ করে, সিস্টেমের স্থানিক সংগঠন বৈশিষ্ট্য প্রদর্শন করে।

উপসংহার এবং আলোচনা

প্রধান উপসংহার

१. Foragax সফলভাবে বড় আকারের মাল্টি-এজেন্ট খাদ্য সংগ্রহ আচরণের দক্ষ সিমুলেশন বাস্তবায়ন করেছে २. JAX ব্যাকএন্ড উল্লেখযোগ্যভাবে গণনামূলক কর্মক্ষমতা এবং স্কেলেবিলিটি উন্নত করেছে ३. ফ্রেমওয়ার্ক বিভিন্ন খাদ্য সংগ্রহ পরিস্থিতি মডেল করার জন্য যথেষ্ট নমনীয়তা প্রদান করে ४. ক্রমাগত অ-পর্যায়ক্রমিক সিমুলেশন আকর্ষণীয় উদীয়মান আচরণ প্রকাশ করতে পারে

সীমাবদ্ধতা

१. পূর্ব-বরাদ্দ সীমাবদ্ধতা: ডেটা কাঠামোর সর্বাধিক আকার পূর্বে জানতে হবে २. মেমরি খরচ: শূন্য প্যাডিং কৌশল মেমরি ব্যবহারের দক্ষতা হ্রাস করতে পারে ३. অ্যালগরিদম পরিসীমা: বর্তমানে প্রধানত স্নায়ু বিবর্তন অ্যালগরিদম সমর্থন করে, শক্তিশালী শেখার সমর্থন উন্নতির অপেক্ষায় ४. যাচাইকরণ অপর্যাপ্ত: প্রকৃত জৈবিক খাদ্য সংগ্রহ আচরণের সাথে পরিমাণগত তুলনার অভাব

ভবিষ্যত দিকনির্দেশনা

१. গ্রেডিয়েন্ট-ভিত্তিক শক্তিশালী শেখার অ্যালগরিদমে সম্প্রসারণ २. মেমরি ব্যবহার দক্ষতা উন্নতি ३. আরও জৈবিক বাস্তবতা যাচাইকরণ যোগ করা ४. আরও সমৃদ্ধ বিশ্লেষণ সরঞ্জাম উন্নয়ন

গভীর মূল্যায়ন

সুবিধা

१. শক্তিশালী প্রযুক্তিগত উদ্ভাবনী: JAX-এর উচ্চ-কর্মক্ষমতা গণনা ক্ষমতা মাল্টি-এজেন্ট মডেলিং ক্ষেত্রে প্রবর্তন করা २. উচ্চ ব্যবহারিক মূল্য: বড় আকারের এজেন্ট সিমুলেশনের মূল বাধা সমস্যা সমাধান করা ३. যুক্তিসঙ্গত ডিজাইন: পাঁচ-স্তরের বিমূর্ততা আর্কিটেকচার স্পষ্ট, সম্প্রসারণ এবং ব্যবহার সহজ ४. উৎকৃষ্ট কর্মক্ষমতা: কার্যকারিতা বজায় রেখে উল্লেখযোগ্য কর্মক্ষমতা উন্নতি অর্জন করা

অসুবিধা

१. সীমিত পরীক্ষামূলক যাচাইকরণ: প্রধানত প্রযুক্তিগত সম্ভাব্যতা প্রদর্শন করে, গভীর আচরণ বিশ্লেষণের অভাব २. অপর্যাপ্ত জৈবিক যাচাইকরণ: প্রকৃত খাদ্য সংগ্রহ আচরণের সাথে পরিমাণগত তুলনা নেই ३. সীমিত তাত্ত্বিক অবদান: প্রধানত প্রকৌশল বাস্তবায়ন, তাত্ত্বিক উদ্ভাবন তুলনামূলকভাবে কম ४. অপর্যাপ্ত ডকুমেন্টেশন: কিছু প্রযুক্তিগত বিবরণ বর্ণনা যথেষ্ট বিস্তারিত নয়

প্রভাব

१. ক্ষেত্র অবদান: মাল্টি-এজেন্ট সিস্টেম গবেষণার জন্য শক্তিশালী গণনা সরঞ্জাম প্রদান করা २. ব্যবহারিক মূল্য: আরও বড় আকারের, আরও জটিল ইকোলজি এবং আচরণ গবেষণা সমর্থন করতে পারে ३. পুনরুৎপাদনযোগ্যতা: খোলা-উৎস কোড গবেষণা পুনরুৎপাদনযোগ্যতা এবং সম্প্রসারণযোগ্যতা বৃদ্ধি করে ४. ক্রস-ডোমেইন প্রয়োগ: যদিও খাদ্য সংগ্রহে ফোকাস করা হয়েছে, তবে অন্যান্য মাল্টি-এজেন্ট পরিস্থিতিতে সম্প্রসারণযোগ্য

প্রযোজ্য পরিস্থিতি

१. ইকোলজি গবেষণা: বড় আকারের প্রাণী খাদ্য সংগ্রহ আচরণ মডেলিং २. বিবর্তনীয় জীববিজ্ঞান: জনসংখ্যা বিবর্তনীয় গতিশীলতা গবেষণা ३. কৃত্রিম বুদ্ধিমত্তা: মাল্টি-এজেন্ট শক্তিশালী শেখার অ্যালগরিদম পরীক্ষা ४. জটিল সিস্টেম: উদীয়মান আচরণ এবং স্ব-সংগঠন ঘটনা গবেষণা ५. সামাজিক বিজ্ঞান: বড় আকারের সামাজিক গতিশীলতা মডেলিং

রেফারেন্স

পত্রটি একাধিক গুরুত্বপূর্ণ ক্ষেত্রের মূল সাহিত্য উদ্ধৃত করে, যার মধ্যে রয়েছে:

JAX লাইব্রেরির প্রযুক্তিগত ডকুমেন্টেশন
খাদ্য সংগ্রহ তত্ত্বের ক্লাসিক কাজ (Stephens & Krebs, 1986)
বিদ্যমান এজেন্ট মডেলিং ফ্রেমওয়ার্ক (Agents.jl, Mesa)
স্নায়ু বিবর্তন এবং শক্তিশালী শেখার সর্বশেষ অগ্রগতি

সামগ্রিক মূল্যায়ন: এটি একটি উচ্চ-মানের সিস্টেম পত্র যা আধুনিক উচ্চ-কর্মক্ষমতা গণনা প্রযুক্তি এবং মাল্টি-এজেন্ট মডেলিংকে একত্রিত করে, এই ক্ষেত্রের একটি গুরুত্বপূর্ণ প্রযুক্তিগত বাধা সমাধান করে। যদিও তাত্ত্বিক উদ্ভাবনে তুলনামূলকভাবে সীমিত, এর প্রকৌশল মূল্য এবং ব্যবহারিকতা এটিকে মাল্টি-এজেন্ট সিস্টেম গবেষণার একটি গুরুত্বপূর্ণ অবদান করে তোলে।